Xuweiyi commited on
Commit
b8ab20d
1 Parent(s): 2c4dfbb

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
  torch==2.1.2
2
  torchvision==0.16.2
3
  torchaudio==2.1.2
4
- pytorch-cuda==11.8
5
  diffusers==0.26.1
6
  transformers==4.25.1
7
  imageio==2.27.0
 
1
  torch==2.1.2
2
  torchvision==0.16.2
3
  torchaudio==2.1.2
4
+ accelerate==0.25.0
5
  diffusers==0.26.1
6
  transformers==4.25.1
7
  imageio==2.27.0