XingyiHe commited on
Commit
396b26a
·
1 Parent(s): 48f422d

Fix ZeroGPU torch compatibility

Browse files
Files changed (2) hide show
  1. README.md +2 -3
  2. requirements.txt +2 -2
README.md CHANGED
@@ -4,9 +4,8 @@ emoji: 🏢
4
  colorFrom: red
5
  colorTo: blue
6
  sdk: gradio
7
- python_version: 3.10.13
8
- sdk_version: 4.44.0
9
- # sdk_version: 5.23.0
10
  app_file: app.py
11
  pinned: false
12
  license: apache-2.0
 
4
  colorFrom: red
5
  colorTo: blue
6
  sdk: gradio
7
+ python_version: "3.10.13"
8
+ sdk_version: "4.44.0"
 
9
  app_file: app.py
10
  pinned: false
11
  license: apache-2.0
requirements.txt CHANGED
@@ -31,8 +31,8 @@ scipy
31
  seaborn
32
  shapely
33
  tensorboardX==2.6.1
34
- torch==2.2.2
35
- torchvision==0.17.2
36
  torchmetrics==0.6.0
37
  tqdm
38
  uvicorn
 
31
  seaborn
32
  shapely
33
  tensorboardX==2.6.1
34
+ torch==2.8.0
35
+ torchvision==0.23.0
36
  torchmetrics==0.6.0
37
  tqdm
38
  uvicorn