pandaphd commited on
Commit
d7ca852
·
1 Parent(s): f658ca0

fix versions

Browse files
Files changed (1) hide show
  1. requirements.txt +15 -18
requirements.txt CHANGED
@@ -1,22 +1,19 @@
1
-
2
- torch==2.2.2
3
- torchvision==0.17.2
4
- torchaudio==2.2.2
5
- xformers==0.0.25.post1
6
- imageio==2.27.0
7
- imageio[ffmpeg]
8
- imageio[pyav]
9
- peft==0.9.0
10
- numpy==1.26.4
11
- ms-swift==2.0.0
12
- accelerate==0.28.0
13
- diffusers==0.11.1
14
- transformers==4.38.2
15
- huggingface_hub==0.25.2
16
- gradio==3.50.2
17
  gdown
18
- triton
19
  einops
 
20
  omegaconf
21
  safetensors
22
- spaces
 
 
 
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu121
2
+ torch==2.1.1
3
+ torchvision==0.16.1
4
+ torchaudio==2.1.1
5
+ diffusers==0.24.0
6
+ imageio==2.36.0
7
+ imageio-ffmpeg
8
+ transformers
9
+ accelerate
10
+ opencv-python
 
 
 
 
 
 
11
  gdown
 
12
  einops
13
+ decord
14
  omegaconf
15
  safetensors
16
+ gradio
17
+ wandb
18
+ triton
19
+ huggingface_hub