matthoffner commited on
Commit
c980911
·
1 Parent(s): 1ec7ff3

Update llm.py

Browse files
Files changed (1) hide show
  1. llm.py +1 -1
llm.py CHANGED
@@ -196,7 +196,7 @@ def ask_ai(
196
  f16_kv=True,
197
  max_tokens=200
198
  )
199
- embeddings = HuggingFaceEmbeddings(model_kwargs={"device": "mps"})
200
  embed_model = LangchainEmbedding(embeddings)
201
  llm_predictor = LLMPredictor(
202
  llm=llm
 
196
  f16_kv=True,
197
  max_tokens=200
198
  )
199
+ embeddings = HuggingFaceEmbeddings()
200
  embed_model = LangchainEmbedding(embeddings)
201
  llm_predictor = LLMPredictor(
202
  llm=llm