Spaces:

zliang
/

ClimateChat

Paused

zliang commited on Sep 30, 2023

Commit

72171c3

•

1 Parent(s): 2fff144

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ import time
 # import e5-large-v2 embedding model
 model_name = "intfloat/e5-large-v2"
-model_kwargs = {'device': 'cuda'}
 encode_kwargs = {'normalize_embeddings': False}
 embeddings = HuggingFaceEmbeddings(
     model_name=model_name,

 # import e5-large-v2 embedding model
 model_name = "intfloat/e5-large-v2"
+model_kwargs = {'device': 'cpu'}
 encode_kwargs = {'normalize_embeddings': False}
 embeddings = HuggingFaceEmbeddings(
     model_name=model_name,