-
Notifications
You must be signed in to change notification settings - Fork 37
Description
Traceback (most recent call last):
File "D:\MachineLearningPractice-main\chatglm_qa\chatglm_document_qa.py", line 69, in
db = store_chroma(documents, embeddings)
File "D:\MachineLearningPractice-main\chatglm_qa\chatglm_document_qa.py", line 59, in store_chroma
db = Chroma.from_documents(docs, embeddings, persist_directory=persist_directory)
File "D:\Anaconda3\lib\site-packages\langchain_community\vectorstores\chroma.py", line 778, in from_documents
return cls.from_texts(
File "D:\Anaconda3\lib\site-packages\langchain_community\vectorstores\chroma.py", line 736, in from_texts
chroma_collection.add_texts(
File "D:\Anaconda3\lib\site-packages\langchain_community\vectorstores\chroma.py", line 275, in add_texts
embeddings = self._embedding_function.embed_documents(texts)
File "D:\Anaconda3\lib\site-packages\langchain_community\embeddings\huggingface.py", line 91, in embed_documents
embeddings = self.client.encode(texts, **self.encode_kwargs)
File "D:\Anaconda3\lib\site-packages\sentence_transformers\SentenceTransformer.py", line 153, in encode
self.to(device)
File "D:\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 989, in to
return self._apply(convert)
File "D:\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 641, in _apply
module._apply(fn)
File "D:\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 641, in _apply
module._apply(fn)
File "D:\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 641, in _apply
module._apply(fn)
[Previous line repeated 1 more time]
File "D:\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 664, in apply
param_applied = fn(param)
File "D:\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 987, in convert
return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking)
File "D:\Anaconda3\lib\site-packages\torch\cuda_init.py", line 221, in _lazy_init
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled