update requirements (#5407)

a28c9715 · Tong Li · GitHub · 0a25e16e · a28c9715
Unverified Commit a28c9715 authored Feb 28, 2024 by Tong Li Committed by GitHub Feb 28, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

applications/Colossal-LLaMA-2/requirements.txt applications/Colossal-LLaMA-2/requirements.txt +2 -2

No files found.
--- a/applications/Colossal-LLaMA-2/requirements.txt
+++ b/applications/Colossal-LLaMA-2/requirements.txt
 torch<2.0.0, >=1.12.1
 packaging==23.1
-colossalai==0.3.2
+colossalai==0.3.5
 autoflake==2.2.1
 black==23.9.1
-transformers
+transformers==4.33.3
 tensorboard==2.14.0
 six==1.16.0
 datasets