zimhe commited on
Commit
51cdcee
·
1 Parent(s): 04a47cb

revise requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -4
requirements.txt CHANGED
@@ -1,13 +1,15 @@
1
- accelerate
2
- numpy
3
- pillow
4
- diffusers
5
  --index-url https://download.pytorch.org/whl/cu121
 
6
  torch==2.4.0+cu121
7
  torchvision
8
  torchaudio
 
 
 
 
9
  transformers
10
  xformers
11
  realesrgan
12
  py360convert
 
13
  https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.4cxx11abiTRUE-cp310-cp310-linux_x86_64.whl
 
 
 
 
 
1
  --index-url https://download.pytorch.org/whl/cu121
2
+ --extra-index-url https://download.pytorch.org/whl/cu121
3
  torch==2.4.0+cu121
4
  torchvision
5
  torchaudio
6
+ accelerate
7
+ numpy
8
+ pillow
9
+ diffusers
10
  transformers
11
  xformers
12
  realesrgan
13
  py360convert
14
+ gradio
15
  https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.4cxx11abiTRUE-cp310-cp310-linux_x86_64.whl