Update requirements.txt
Browse files- requirements.txt +4 -3
requirements.txt
CHANGED
|
@@ -2,14 +2,15 @@ torch
|
|
| 2 |
torchvision
|
| 3 |
torchaudio
|
| 4 |
transformers>=4.25.1
|
| 5 |
-
gradio
|
|
|
|
|
|
|
| 6 |
ftfy
|
| 7 |
tensorboard
|
| 8 |
datasets
|
| 9 |
Pillow==9.5.0
|
| 10 |
opencv-python
|
| 11 |
-
imgaug
|
| 12 |
-
accelerate==0.20.3
|
| 13 |
image-reward
|
| 14 |
hpsv2
|
| 15 |
torchmetrics
|
|
|
|
| 2 |
torchvision
|
| 3 |
torchaudio
|
| 4 |
transformers>=4.25.1
|
| 5 |
+
gradio>=3.50.0,<3.51
|
| 6 |
+
accelerate>=0.26.0
|
| 7 |
+
fsspec[http]==2024.6.1
|
| 8 |
ftfy
|
| 9 |
tensorboard
|
| 10 |
datasets
|
| 11 |
Pillow==9.5.0
|
| 12 |
opencv-python
|
| 13 |
+
imgaug
|
|
|
|
| 14 |
image-reward
|
| 15 |
hpsv2
|
| 16 |
torchmetrics
|