Spaces:
Sleeping
Sleeping
embeddings test-4
Browse files
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="
|
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")
|