Alessio Grancini commited on
Commit
2d8c22b
·
verified ·
1 Parent(s): 865e874

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -3
requirements.txt CHANGED
@@ -10,9 +10,8 @@ tqdm>=4.64.0
10
  urllib3>=1.26.14
11
 
12
  # ZeroGPU specific requirements
13
- --extra-index-url https://download.pytorch.org/whl/cu118
14
- torch==2.0.1+cu118
15
- torchvision==0.15.2+cu118
16
  gradio==4.44.1 # Specific version for ZeroGPU compatibility
17
  spaces
18
 
 
10
  urllib3>=1.26.14
11
 
12
  # ZeroGPU specific requirements
13
+ torch==2.1.2 # Specific ZeroGPU supported version
14
+ torchvision==0.16.2 # Compatible with torch 2.1.2
 
15
  gradio==4.44.1 # Specific version for ZeroGPU compatibility
16
  spaces
17