RCaz commited on
Commit
835cd89
·
verified ·
1 Parent(s): bd3055b

change init model to use cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ print("load embeddings")
43
  embedding_model = HuggingFaceEmbeddings(
44
  model_name="intfloat/e5-base-v2",
45
  # multi_process=True,
46
- model_kwargs={"device": "cuda"}, # use cuda for faster embeddings on nbidia GPUs
47
  encode_kwargs={"normalize_embeddings": True}, # Set `True` for cosine similarity
48
  )
49
  print("load vector store")
 
43
  embedding_model = HuggingFaceEmbeddings(
44
  model_name="intfloat/e5-base-v2",
45
  # multi_process=True,
46
+ model_kwargs={"device": "cpu"}, # use cuda for faster embeddings on nbidia GPUs
47
  encode_kwargs={"normalize_embeddings": True}, # Set `True` for cosine similarity
48
  )
49
  print("load vector store")