Spaces:

sandaruths
/

app1

Sleeping

Sandaruth commited on Feb 18

Commit

47a9554

•

1 Parent(s): 9dbd819

Add model.py file

Files changed (1) hide show

model.py CHANGED Viewed

@@ -22,7 +22,7 @@ model_name = "BAAI/bge-large-en"
 embedding = HuggingFaceBgeEmbeddings(
     model_name = model_name,
-    model_kwargs = {'device':'cuda'},
     encode_kwargs = {'normalize_embeddings': True}
 )

 embedding = HuggingFaceBgeEmbeddings(
     model_name = model_name,
+    # model_kwargs = {'device':'cuda'},
     encode_kwargs = {'normalize_embeddings': True}
 )