Update requirements.txt

requirements.txt

@@ -1,13 +1,19 @@
-torch
+torch==2.6.0
 torchvision
-transformers>=4.41.0
-accelerate
-pillow
-gradio
+transformers==4.46.3
+tokenizers==0.20.3
+sentencepiece>=0.1.99
+accelerate>=0.24.0
 huggingface_hub>=0.23.0
-hf_transfer
-timm
+safetensors>=0.4.0
+pillow>=10.0.0
+gradio>=4.0.0
+spaces>=0.20.0
+protobuf>=3.20.0
 einops
-safetensors
+timm
 addict
+easydict
 matplotlib
+hf_transfer
+flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn-2.7.3+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl