diff --git a/applications/ColossalChat/requirements.txt b/applications/ColossalChat/requirements.txt index 82586646c..ef3a5a0e8 100755 --- a/applications/ColossalChat/requirements.txt +++ b/applications/ColossalChat/requirements.txt @@ -1,6 +1,6 @@ transformers>=4.36.2 tqdm -datasets +datasets==2.14.7 loralib colossalai>=0.3.7 torch>=1.12.1