chunzhi/chatgpt_academic
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add installation info

Browse Source
This commit is contained in:
qingxu fu 2023-12-07 21:29:41 +08:00
parent 7bac8f4bd3
commit 21bccf69d2
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
crazy_functions/vector_fns/vector_database.py
View File

@ -199,8 +199,7 @@ class LocalDocQA:
self.vector_store.save_local(vs_path) self.vector_store.save_local(vs_path)
return vs_path, loaded_files return vs_path, loaded_files
else: else:
self.vector_store = FAISS.load_local(vs_path, text2vec) raise RuntimeError("文件加载失败，请检查文件格式是否正确")
return vs_path, loaded_files
def get_loaded_file(self, vs_path): def get_loaded_file(self, vs_path):
ds = self.vector_store.docstore ds = self.vector_store.docstore

11
crazy_functions/知识库问答.py
View File

@ -2,9 +2,14 @@ from toolbox import CatchException, update_ui, ProxyNetworkActivate, update_ui_l
from .crazy_utils import request_gpt_model_in_new_thread_with_ui_alive, get_files_from_everything from .crazy_utils import request_gpt_model_in_new_thread_with_ui_alive, get_files_from_everything
install_msg =""" install_msg ="""
pip3 install torch --index-url https://download.pytorch.org/whl/cpu
pip3 install transformers --upgrade 1. python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
pip3 install langchain sentence-transformers unstructured[all-docs] faiss-cpu nltk beautifulsoup4 bitsandbytes tabulate icetk --upgrade
2. python -m pip install transformers protobuf langchain sentence-transformers faiss-cpu nltk beautifulsoup4 bitsandbytes tabulate icetk --upgrade
3. python -m pip install unstructured[all-docs] --upgrade
4. python -c 'import nltk; nltk.download("punkt")'
""" """
@CatchException @CatchException