RAG_SPACE / pre_download.py
NimrodDev's picture
cmc
e95e29f
raw
history blame contribute delete
329 Bytes
from sentence_transformers import SentenceTransformer
if __name__ == "__main__":
print("🔹 Pre-downloading embedding model: all-MiniLM-L6-v2 ...")
model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
model.encode(["test"]) # trigger download
print("✅ Embedding model cached successfully.")