AFischer1985 commited on
Commit
923df80
·
verified ·
1 Parent(s): 3851ace

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -40,7 +40,7 @@ from chromadb.utils import embedding_functions
40
  default_ef = embedding_functions.DefaultEmbeddingFunction()
41
  #sentence_transformer_ef = embedding_functions.SentenceTransformerEmbeddingFunction(model_name="T-Systems-onsite/cross-en-de-roberta-sentence-transformer")
42
  #instructor_ef = embedding_functions.InstructorEmbeddingFunction(model_name="hkunlp/instructor-large", device="cuda")
43
- embeddingModel = embedding_functions.SentenceTransformerEmbeddingFunction(model_name="T-Systems-onsite/cross-en-de-roberta-sentence-transformer", device="cuda")
44
 
45
  print(str(client.list_collections()))
46
 
 
40
  default_ef = embedding_functions.DefaultEmbeddingFunction()
41
  #sentence_transformer_ef = embedding_functions.SentenceTransformerEmbeddingFunction(model_name="T-Systems-onsite/cross-en-de-roberta-sentence-transformer")
42
  #instructor_ef = embedding_functions.InstructorEmbeddingFunction(model_name="hkunlp/instructor-large", device="cuda")
43
+ embeddingModel = embedding_functions.SentenceTransformerEmbeddingFunction(model_name="T-Systems-onsite/cross-en-de-roberta-sentence-transformer") #, device="cuda")
44
 
45
  print(str(client.list_collections()))
46