diff --git a/requirements.txt b/requirements.txt index 295e7430..3efd0200 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,9 +7,9 @@ xformers>=0.0.22.post4 openai>=0.28.1 sentence_transformers transformers>=4.34 -torch>=2.0.1 # suggest version 2.1 -torchvision -torchaudio +torch>=2.0.1 # suggest version 2.1,#on win, install the cuda version manually if you want use gpu +torchvision #on win, install the cuda version manually if you want use gpu +torchaudio #on win, install the cuda version manually if you want use gpu fastapi>=0.104 nltk>=3.8.1 uvicorn~=0.23.1 @@ -53,7 +53,7 @@ vllm>=0.2.0; sys_platform == "linux" # 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-antd-components>=0.1.11 streamlit-chatbox>=1.1.11