thewhole commited on
Commit
4d3d631
·
1 Parent(s): 2fa4776

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -2
requirements.txt CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  lightning==2.0.0
2
  omegaconf==2.3.0
3
  jaxtyping
@@ -26,7 +29,7 @@ git+https://github.com/ashawkey/envlight.git
26
  # torchmetrics
27
 
28
  # deepfloyd
29
- xformers
30
  bitsandbytes==0.38.1
31
  sentencepiece
32
  safetensors
@@ -39,4 +42,4 @@ taming-transformers-rom1504
39
  git+https://github.com/openai/CLIP.git
40
 
41
  #controlnet
42
- controlnet_aux
 
1
+ torch==2.0.1+cu117
2
+ torchvision==0.15.2+cu117
3
+ torchaudio==2.0.2
4
  lightning==2.0.0
5
  omegaconf==2.3.0
6
  jaxtyping
 
29
  # torchmetrics
30
 
31
  # deepfloyd
32
+ xformers==0.0.20
33
  bitsandbytes==0.38.1
34
  sentencepiece
35
  safetensors
 
42
  git+https://github.com/openai/CLIP.git
43
 
44
  #controlnet
45
+ controlnet_aux