diff --git a/applications/ColossalChat/requirements.txt b/applications/ColossalChat/requirements.txt index c5457c22c..ef3a5a0e8 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.8 +colossalai>=0.3.7 torch>=1.12.1 langchain tokenizers