Update requirements.txt

针对win上的reqirements某些包的安装安装增说明
2023-11-25 19:34:57 +08:00 · 2023-11-25 19:34:57 +08:00 · 3c4d4a5345
parent 3839c212db
commit 3c4d4a5345
1 changed files with 4 additions and 4 deletions
--- 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