ZYMPKU commited on
Commit
1820a80
1 Parent(s): 6497501
Files changed (1) hide show
  1. requirements.txt +3 -0
requirements.txt CHANGED
@@ -1,7 +1,9 @@
1
  colorlover==0.3.0
 
2
  gradio==3.41.0
3
  imageio==2.31.2
4
  img2dataset==1.42.0
 
5
  lpips==0.1.4
6
  matplotlib==3.7.2
7
  numpy==1.25.1
@@ -16,6 +18,7 @@ scikit-learn==1.3.0
16
  scipy==1.11.1
17
  seaborn==0.12.2
18
  tensorboard==2.14.0
 
19
  tokenizers==0.13.3
20
  torch==2.1.0
21
  torchvision==0.16.0
 
1
  colorlover==0.3.0
2
+ einops==0.6.1
3
  gradio==3.41.0
4
  imageio==2.31.2
5
  img2dataset==1.42.0
6
+ kornia==0.6.9
7
  lpips==0.1.4
8
  matplotlib==3.7.2
9
  numpy==1.25.1
 
18
  scipy==1.11.1
19
  seaborn==0.12.2
20
  tensorboard==2.14.0
21
+ timm==0.9.2
22
  tokenizers==0.13.3
23
  torch==2.1.0
24
  torchvision==0.16.0