embedding-model / requirements.txt
IotaCluster's picture
Update requirements.txt
0cf4c45 verified
raw
history blame
301 Bytes
# Gradio UI
gradio>=3.0
# Dense embeddings
sentence-transformers>=2.2.2
torch>=1.12.0
# Sparse BM25
rank_bm25>=0.2.2
# Late-interaction (ColBERT)
transformers>=4.26.0
colbert-py>=1.0.0 # or replace with your ColBERT package name/version
# (Optional) HF hub support
huggingface-hub>=0.13.0