JiantaoLin commited on
Commit
6927287
·
1 Parent(s): eb90cd3
Files changed (1) hide show
  1. requirements.txt +3 -1
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
  torch==2.1.0
2
  torchvision==0.16.0
 
3
  -f https://download.pytorch.org/whl/cu121
4
 
5
  comfy==0.0.1
@@ -38,4 +39,5 @@ ninja
38
  sentencepiece
39
  accelerate
40
  git+https://github.com/huggingface/diffusers
41
- einops
 
 
1
  torch==2.1.0
2
  torchvision==0.16.0
3
+ torchaudio==2.1.0
4
  -f https://download.pytorch.org/whl/cu121
5
 
6
  comfy==0.0.1
 
39
  sentencepiece
40
  accelerate
41
  git+https://github.com/huggingface/diffusers
42
+ einops
43
+ numpy==1.26.3