lexRetriver / download_model.py
Amogh Gupta
HF Fix: Pre-download embedding model to avoid runtime connection errors
4905a93
raw
history blame contribute delete
278 Bytes
from sentence_transformers import SentenceTransformer
import os
model_name = "BAAI/bge-small-en-v1.5"
save_path = "./models/bge-small"
print(f"Pre-downloading {model_name}...")
model = SentenceTransformer(model_name)
model.save(save_path)
print(f"Model saved to {save_path}")