chongzhou commited on
Commit
c02ab21
·
1 Parent(s): bd2518a

Pin torch==2.10.0/torchvision==0.25.0 for ZeroGPU

Browse files

ZeroGPU supports torch 2.8.0/2.9.1/2.10.0/2.11.0; 2.4.0 was rejected.
2.10.0 is supported and matches the known-good local env; 0.25.0 is its
paired torchvision.

Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
- torch==2.4.0
2
- torchvision==0.19.0
3
  numpy>=1.24.4
4
  tqdm>=4.66.1
5
  hydra-core>=1.3.2
 
1
+ torch==2.10.0
2
+ torchvision==0.25.0
3
  numpy>=1.24.4
4
  tqdm>=4.66.1
5
  hydra-core>=1.3.2