Merge pull request #2170 from chatchat-space/hzg0601-patch-2

Update requirements.txt
This commit is contained in:
Zhi-guo Huang 2023-11-25 19:35:11 +08:00 committed by GitHub
commit 63bdca1343
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions

View File

@ -7,9 +7,9 @@ xformers>=0.0.22.post4
openai>=0.28.1 openai>=0.28.1
sentence_transformers sentence_transformers
transformers>=4.34 transformers>=4.34
torch>=2.0.1 # suggest version 2.1 torch>=2.0.1 # suggest version 2.1,#on win, install the cuda version manually if you want use gpu
torchvision torchvision #on win, install the cuda version manually if you want use gpu
torchaudio torchaudio #on win, install the cuda version manually if you want use gpu
fastapi>=0.104 fastapi>=0.104
nltk>=3.8.1 nltk>=3.8.1
uvicorn~=0.23.1 uvicorn~=0.23.1
@ -53,7 +53,7 @@ vllm>=0.2.0; sys_platform == "linux"
# WebUI requirements # WebUI requirements
streamlit~=1.27.0 streamlit~=1.27.0 # on win, make sure write its path in environment variable
streamlit-option-menu>=0.3.6 streamlit-option-menu>=0.3.6
streamlit-antd-components>=0.1.11 streamlit-antd-components>=0.1.11
streamlit-chatbox>=1.1.11 streamlit-chatbox>=1.1.11