Update requirements

2025-01-07 15:30:28 +08:00
parent fbc0981546
commit ff933b7563
1 changed files with 6 additions and 2 deletions
@@ -2,5 +2,9 @@ numpy
 Pillow
 Requests
 flash_attn
-transformers
-einops
+accelerate>=0.25.0
+huggingface-hub
+transformers>=4.44.0
+einops>=0.7.0
+torch>=2.1.2
+torchvision>=0.16.2