fix torch colossalai version

2024-06-28 03:58:29 +00:00 · 2024-06-28 03:58:29 +00:00 · a8af6ccb73
parent e7527762a1
commit a8af6ccb73
2 changed files with 3 additions and 3 deletions
--- a/applications/ColossalChat/examples/requirements.txt
+++ b/applications/ColossalChat/examples/requirements.txt
@ -1,4 +1,4 @@
 pandas>=1.4.1
 sentencepiece
-colossalai
+colossalai==0.4.0
 prompt_toolkit
--- a/applications/ColossalChat/requirements.txt
+++ b/applications/ColossalChat/requirements.txt
@ -2,8 +2,8 @@ transformers==4.36.2
 tqdm
 datasets==2.14.7
 loralib
-colossalai>=0.3.7
-torch>=1.12.1
+colossalai==0.4.0
+torch>=2.1.0
 langchain
 tokenizers
 fastapi