Chris4K commited on
Commit
41494a0
1 Parent(s): 59dbb0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -426,7 +426,7 @@ def get_embedding_model(model_type, model_name):
426
  return HuggingFaceEmbeddings(
427
  model_name=model_path,
428
  multi_process=True,
429
- model_kwargs={"device": "cuda"},
430
  #encode_kwargs={"normalize_embeddings": True}, # Set `True` for cosine similarity
431
  )
432
  elif model_type == 'OpenAI':
 
426
  return HuggingFaceEmbeddings(
427
  model_name=model_path,
428
  multi_process=True,
429
+ # model_kwargs={"device": "cpu"},
430
  #encode_kwargs={"normalize_embeddings": True}, # Set `True` for cosine similarity
431
  )
432
  elif model_type == 'OpenAI':