Divyanshh commited on
Commit
5980152
·
verified ·
1 Parent(s): 1abb7b4

Embeddings model test -3

Browse files
Files changed (1) hide show
  1. util.py +1 -1
util.py CHANGED
@@ -22,7 +22,7 @@ genai.configure(api_key = os.environ['GOOGLE_API_KEY'])
22
 
23
 
24
  model_kwargs = {'device': 'cpu'}
25
- embeddings = HuggingFaceEmbeddings(model_name="Salesforce/codet5p-110m-embedding",model_kwargs=model_kwargs, )
26
  # embeddings = SentenceTransformer(model_name_or_path="All-MiniLM-L6-v2")
27
 
28
  # tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
 
22
 
23
 
24
  model_kwargs = {'device': 'cpu'}
25
+ embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2",model_kwargs=model_kwargs, )
26
  # embeddings = SentenceTransformer(model_name_or_path="All-MiniLM-L6-v2")
27
 
28
  # tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")