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
|