ifmain commited on
Commit
4ba36d4
1 Parent(s): fc1525b

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu118
2
- torch
3
  git+https://github.com/huggingface/diffusers.git
4
  transformers
5
  accelerate
 
1
  --extra-index-url https://download.pytorch.org/whl/cu118
2
+ torch torchvision
3
  git+https://github.com/huggingface/diffusers.git
4
  transformers
5
  accelerate