MedRAG / scripts /preload_models.py
hetsheta's picture
Add preload_models.py script (force-add past gitignore)
9a7b546
Raw
History Blame Contribute Delete
348 Bytes
from sentence_transformers import SentenceTransformer
print("Downloading PubMedBERT embedding model...")
SentenceTransformer("NeuML/pubmedbert-base-embeddings")
print("Done.")
print("Downloading cross-encoder reranker model...")
SentenceTransformer("cross-encoder/ms-marco-MiniLM-L-6-v2")
print("Done.")
print("All models cached successfully.")