flamehaze1115 commited on
Commit
d1214ac
1 Parent(s): 2453164

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -1,7 +1,8 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu113
2
- torch==1.12.1
3
  torchvision==0.13.1
4
  diffusers[torch]==0.19.3
 
5
  transformers>=4.25.1
6
  bitsandbytes==0.35.4
7
  decord==0.6.0
@@ -27,4 +28,4 @@ torch_efficient_distloss
27
  tensorboard
28
  rembg
29
  segment_anything
30
- xformers
 
1
  --extra-index-url https://download.pytorch.org/whl/cu113
2
+ torch==1.13.1
3
  torchvision==0.13.1
4
  diffusers[torch]==0.19.3
5
+ xformers==0.0.16
6
  transformers>=4.25.1
7
  bitsandbytes==0.35.4
8
  decord==0.6.0
 
28
  tensorboard
29
  rembg
30
  segment_anything
31
+