yiren98 commited on
Commit
737d7ff
·
verified ·
1 Parent(s): c04b2ae

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -43,5 +43,7 @@ numpy<=2.0
43
  rich==13.7.0
44
  # for T5XXL tokenizer (SD3/FLUX)
45
  sentencepiece==0.2.0
 
 
46
  # for kohya_ss library
47
  -e git+https://github.com/showlab/MakeAnything.git#egg=library
 
43
  rich==13.7.0
44
  # for T5XXL tokenizer (SD3/FLUX)
45
  sentencepiece==0.2.0
46
+ torch==2.5.1+cu124
47
+ torchvision==0.20.1+cu124
48
  # for kohya_ss library
49
  -e git+https://github.com/showlab/MakeAnything.git#egg=library