chrisjcc commited on
Commit
3990b0e
·
verified ·
1 Parent(s): 37a666c

Update torch version

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
  diffusers==0.34.0
2
  gradio==4.44.0
3
- torch==2.4.1
4
- transformers>=4.44.0
 
1
  diffusers==0.34.0
2
  gradio==4.44.0
3
+ torch>=2.6 --index-url https://download.pytorch.org/whl/cu128
4
+ transformers==4.44.0