ReubenSun commited on
Commit
8a0aa9d
·
1 Parent(s): ec0f666
Files changed (1) hide show
  1. requirements.txt +3 -0
requirements.txt CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  datasets==2.19.0
2
  diffusers==0.32.2
3
  einops==0.8.0
 
1
+ torch==2.5.1+cu124
2
+ torchvision==0.20.1+cu124
3
+ torchaudio==2.5.1+cu124
4
  datasets==2.19.0
5
  diffusers==0.32.2
6
  einops==0.8.0