Clement Vachet commited on
Commit
afc2218
·
1 Parent(s): b347a6f

Comment out cpu embedding parameter

Browse files
Files changed (1) hide show
  1. indexing.py +1 -1
indexing.py CHANGED
@@ -68,7 +68,7 @@ def create_db(splits, collection_name):
68
  embedding = HuggingFaceEmbeddings(
69
  model_name="sentence-transformers/paraphrase-multilingual-mpnet-base-v2",
70
  # model_name="sentence-transformers/all-MiniLM-L6-v2",
71
- model_kwargs={"device": "cpu"},
72
  # encode_kwargs={'normalize_embeddings': False}
73
  )
74
  chromadb.api.client.SharedSystemClient.clear_system_cache()
 
68
  embedding = HuggingFaceEmbeddings(
69
  model_name="sentence-transformers/paraphrase-multilingual-mpnet-base-v2",
70
  # model_name="sentence-transformers/all-MiniLM-L6-v2",
71
+ # model_kwargs={"device": "cpu"},
72
  # encode_kwargs={'normalize_embeddings': False}
73
  )
74
  chromadb.api.client.SharedSystemClient.clear_system_cache()