SUPERSHANKY commited on
Commit
99ec559
·
1 Parent(s): 1839e9c

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +17 -0
requirements.txt CHANGED
@@ -19,3 +19,20 @@ torchvision==0.14.1
19
  transformers==4.26.1
20
  xformers==0.0.16
21
  yapf==0.32.0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
19
  transformers==4.26.1
20
  xformers==0.0.16
21
  yapf==0.32.0
22
+
23
+ --extra-index-url https://download.pytorch.org/whl/cu117
24
+ torch
25
+ torchvision==0.14.1+cu117
26
+ #diffusers
27
+ git+https://github.com/huggingface/diffusers.git
28
+ #transformers
29
+ git+https://github.com/huggingface/transformers
30
+ scipy
31
+ ftfy
32
+ psutil
33
+ accelerate==0.16.0
34
+ #OmegaConf
35
+ #pytorch_lightning
36
+ triton==2.0.0.dev20230208
37
+ #https://github.com/apolinario/xformers/releases/download/0.0.3/xformers-0.0.14.dev0-cp38-cp38-linux_x86_64.whl
38
+ 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