diff --git a/train/requirements.txt b/train/requirements.txt index c5260b92..1c01abf8 100644 --- a/train/requirements.txt +++ b/train/requirements.txt @@ -3,8 +3,8 @@ bitsandbytes==0.37.1 datasets==2.10.1 fire==0.5.0 huggingface_hub==0.13.1 -torch==1.13.0 -tqdm==4.65.0 -transformers==4.28.1 -deepspeed==0.9.0 +torch==2.8.0 +tqdm==4.66.3 +transformers==4.53.0 +deepspeed==0.15.1 gradio \ No newline at end of file