coralLight commited on
Commit
341670a
·
1 Parent(s): 39a8866

add inference

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -3
requirements.txt CHANGED
@@ -1,12 +1,13 @@
1
  tqdm
2
  einops
3
  pytorch_lightning
4
- accelerate
5
  torchsde
6
  pycocotools
7
- diffusers
8
  timm
9
- transformers
 
10
  opencv-python
11
  omegaconf
12
  gradio
 
1
  tqdm
2
  einops
3
  pytorch_lightning
4
+ accelerate>=0.20.0
5
  torchsde
6
  pycocotools
7
+ diffusers== 0.32.2
8
  timm
9
+ transformers==4.49
10
+ torch>=2.0.0
11
  opencv-python
12
  omegaconf
13
  gradio