SUPERSHANKY commited on
Commit
948e1ae
1 Parent(s): 18c119a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -5
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu116
2
  torch
3
- torchvision==0.14.1+cu116
4
  #diffusers
5
  git+https://github.com/huggingface/diffusers.git
6
  #transformers
@@ -8,9 +8,9 @@ git+https://github.com/huggingface/transformers
8
  scipy
9
  ftfy
10
  psutil
11
- accelerate==0.15.0
12
  #OmegaConf
13
  #pytorch_lightning
14
- triton==2.0.0.dev20221202
15
  #https://github.com/apolinario/xformers/releases/download/0.0.3/xformers-0.0.14.dev0-cp38-cp38-linux_x86_64.whl
16
- https://github.com/ZyCromerZ/xformers_builds/releases/download/xformers-2023-01-17-Py-3.8-Cuda-11.8.0-PyTorch-1.13.1%2Bcu117-ubuntu-18.04/xformers-0.0.16+6f3c20f.d20230117-cp38-cp38-linux_x86_64.whl
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu117
2
  torch
3
+ torchvision==0.14.1+cu117
4
  #diffusers
5
  git+https://github.com/huggingface/diffusers.git
6
  #transformers
 
8
  scipy
9
  ftfy
10
  psutil
11
+ accelerate==0.16.0
12
  #OmegaConf
13
  #pytorch_lightning
14
+ triton==2.0.0.dev20230208
15
  #https://github.com/apolinario/xformers/releases/download/0.0.3/xformers-0.0.14.dev0-cp38-cp38-linux_x86_64.whl
16
+ https://github.com/ZyCromerZ/xformers_builds/releases/download/xformers-2023-02-12-Py-3.8-Cuda-11.8.0-PyTorch-1.13.1%2Bcu117-ubuntu-18.04/xformers-0.0.17+12c076d.d20230212-cp38-cp38-linux_x86_64.whl