fix colossalai, transformers version

5 months ago · 384c64057d
parent 8aad064fe7
commit 384c64057d
1 changed files with 1 additions and 1 deletions
--- a/applications/ColossalChat/requirements.txt
+++ b/applications/ColossalChat/requirements.txt
@ -2,7 +2,7 @@ transformers>=4.36.2
 tqdm
 datasets==2.14.7
 loralib
-colossalai>=0.3.7
+colossalai==0.3.8
 torch>=1.12.1
 langchain
 tokenizers