Yw22 commited on
Commit
9913e3f
1 Parent(s): 0ea700d
Files changed (1) hide show
  1. requirements.txt +7 -4
requirements.txt CHANGED
@@ -1,15 +1,15 @@
1
  torch
2
  torchvision
3
  torchaudio
4
- transformers>=4.25.1
5
- gradio==3.50.0
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
@@ -22,4 +22,7 @@ scipy
22
  tqdm
23
  einops
24
  diffusers==0.28.0
25
- omegaconf
 
 
 
 
1
  torch
2
  torchvision
3
  torchaudio
4
+ transformers==4.32.1
5
+ gradio==3.50.2
6
  ftfy
7
  tensorboard
8
  datasets
9
  Pillow==9.5.0
10
  opencv-python
11
  imgaug
12
+ accelerate==0.23.0
13
  image-reward
14
  hpsv2
15
  torchmetrics
 
22
  tqdm
23
  einops
24
  diffusers==0.28.0
25
+ omegaconf
26
+ xformers==0.0.26.post1
27
+ scikit-image
28
+ scikit-learn