Personal-AI / model_download.py
AmmarFahmy
from local
de81719
raw
history blame contribute delete
605 Bytes
# from transformers import AutoModel, AutoTokenizer
# from langchain.embeddings import HuggingFaceEmbeddings
# local_model_path = "models"
# # Download the model and tokenizer
# model_name = "sentence-transformers/all-mpnet-base-v2"
# AutoModel.from_pretrained(model_name, cache_dir=local_model_path)
# AutoTokenizer.from_pretrained(model_name, cache_dir=local_model_path)
from sentence_transformers import SentenceTransformer
local_model_path = "model"
# Download the model
model_name = "sentence-transformers/all-mpnet-base-v2"
model = SentenceTransformer(model_name, cache_folder=local_model_path)