SUPERSHANKY commited on
Commit
e08263a
1 Parent(s): dd8de16

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -3
requirements.txt CHANGED
@@ -1,6 +1,7 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu113
2
  torch
3
- torchvision==0.13.1+cu113
 
4
  #diffusers
5
  git+https://github.com/huggingface/diffusers.git
6
  #transformers
@@ -13,4 +14,4 @@ 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-2022-12-10-Py-3.8-Cuda-11.3.1-PyTorch-1.12.1%2Bcu113-ubuntu-18.04/xformers-0.0.15.dev0+affe4da.d20221210-cp38-cp38-linux_x86_64.whl
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu117
2
  torch
3
+ https://download.pytorch.org/whl/cu117/torch-1.13.1%2Bcu117-cp38-cp38-linux_x86_64.whl
4
+ torchvision==0.14.1+cu117
5
  #diffusers
6
  git+https://github.com/huggingface/diffusers.git
7
  #transformers
 
14
  pytorch_lightning
15
  triton==2.0.0.dev20230208
16
  #https://github.com/apolinario/xformers/releases/download/0.0.3/xformers-0.0.14.dev0-cp38-cp38-linux_x86_64.whl
17
+ 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