Divyanshh commited on
Commit
083629c
1 Parent(s): 5980152

embeddings test-4

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="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")
 
22
 
23
 
24
  model_kwargs = {'device': 'cpu'}
25
+ embeddings = HuggingFaceEmbeddings(model_name="michaelfeil/ct2fast-e5-small",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")