radames commited on
Commit
0b89618
1 Parent(s): b53d2ad

requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +12 -1
requirements.txt CHANGED
@@ -4,4 +4,15 @@ numpy
4
  opencv-python
5
  scikit-image
6
  psutil
7
- safetensors
 
 
 
 
 
 
 
 
 
 
 
 
4
  opencv-python
5
  scikit-image
6
  psutil
7
+ torchsde
8
+ torchvision
9
+ einops
10
+ transformers>=4.25.1
11
+ safetensors>=0.3.0
12
+ aiohttp
13
+ pyyaml
14
+ Pillow
15
+ scipy
16
+ tqdm
17
+ psutil
18
+ kornia>=0.7.1