Update requirements.txt
Browse files- requirements.txt +3 -2
requirements.txt
CHANGED
|
@@ -1,8 +1,9 @@
|
|
| 1 |
transformers>=4.46.0,<5
|
| 2 |
accelerate>=0.33.0
|
| 3 |
gradio>=4.44.0
|
| 4 |
-
torch
|
|
|
|
| 5 |
pillow>=10.3.0
|
| 6 |
packaging>=24.0
|
| 7 |
safetensors>=0.4.3
|
| 8 |
-
einops>=0.7.0
|
|
|
|
| 1 |
transformers>=4.46.0,<5
|
| 2 |
accelerate>=0.33.0
|
| 3 |
gradio>=4.44.0
|
| 4 |
+
torch==2.3.1
|
| 5 |
+
torchvision==0.18.1
|
| 6 |
pillow>=10.3.0
|
| 7 |
packaging>=24.0
|
| 8 |
safetensors>=0.4.3
|
| 9 |
+
einops>=0.7.0
|