ai-teaching-assistant / retrieval.py

Commit History

fix input features to be FP16, as assumed for CPU
3842618

kastan commited on

remove fancy CUDA device selection, for plain 'cuda'
421e169

kastan commited on

remove unnecessary transformers dependency
4bada45

kastan commited on

remove unnecessary dot_env dependency
41ae44b

kastan commited on

added Pinecone text retrieval, CLIP text-to-image retrieval, and CLIP image-to-image retrieval
36cfe46

kastan commited on