loose

7 months ago · f5b35c86bc
1 changed files with 2 additions and 2 deletions
--- a/applications/Colossal-LLaMA/requirements.txt
+++ b/applications/Colossal-LLaMA/requirements.txt
@ -1,5 +1,5 @@
 torch==2.1.2
-huggingface-hub==0.22.2
+huggingface-hub
 packaging==24.0
 colossalai==0.3.6
 autoflake==2.2.1
@ -7,7 +7,7 @@ black==23.9.1
 transformers==4.34.1
 tensorboard==2.14.0
 six==1.16.0
-datasets==2.19.0
+datasets
 ninja==1.11.1
 flash-attn>=2.0.0,<=2.0.5
 tqdm