diff --git a/applications/ColossalChat/requirements.txt b/applications/ColossalChat/requirements.txt index ef3a5a0e8..c5457c22c 100755 --- 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