lzyhha commited on
Commit
333b37a
·
1 Parent(s): 1d429c0
Files changed (1) hide show
  1. requirements.txt +4 -2
requirements.txt CHANGED
@@ -1,10 +1,12 @@
 
1
  torch==2.1.0
2
  torchvision==0.16.0
3
  numpy<2
4
- diffusers==0.29.0
 
 
5
  huggingface-hub==0.25.0
6
  tensorboard
7
- transformers
8
  gradio
9
  torchdiffeq
10
  click
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu124
2
  torch==2.1.0
3
  torchvision==0.16.0
4
  numpy<2
5
+ diffusers==0.32.1
6
+ accelerate==1.1.1
7
+ transformers==4.46.2
8
  huggingface-hub==0.25.0
9
  tensorboard
 
10
  gradio
11
  torchdiffeq
12
  click