File tree Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -97,12 +97,12 @@ def python_is_compatible():
97
97
NIGHTLY_VERSION = "dev20240901"
98
98
99
99
# The pip repository that hosts nightly torch packages.
100
- TORCH_NIGHTLY_URL = "https://download.pytorch.org/whl/nightly /cpu"
100
+ TORCH_URL = "https://download.pytorch.org/whl/test /cpu"
101
101
102
102
# pip packages needed by exir.
103
103
EXIR_REQUIREMENTS = [
104
- f "torch==2.5.0. { NIGHTLY_VERSION } " ,
105
- f "torchvision==0.20.0. { NIGHTLY_VERSION } " , # For testing.
104
+ "torch==2.5.0" ,
105
+ "torchvision==0.20.0" , # For testing.
106
106
"typing-extensions" ,
107
107
]
108
108
@@ -121,7 +121,7 @@ def python_is_compatible():
121
121
# TODO: Make each example publish its own requirements.txt
122
122
EXAMPLES_REQUIREMENTS = [
123
123
"timm==1.0.7" ,
124
- f "torchaudio==2.5.0. { NIGHTLY_VERSION } " ,
124
+ "torchaudio==2.5.0" ,
125
125
"torchsr==1.0.4" ,
126
126
"transformers==4.42.4" ,
127
127
]
@@ -140,7 +140,7 @@ def python_is_compatible():
140
140
"install" ,
141
141
* REQUIREMENTS_TO_INSTALL ,
142
142
"--extra-index-url" ,
143
- TORCH_NIGHTLY_URL ,
143
+ TORCH_URL ,
144
144
],
145
145
check = True ,
146
146
)
@@ -167,7 +167,7 @@ def python_is_compatible():
167
167
"--no-build-isolation" ,
168
168
"-v" ,
169
169
"--extra-index-url" ,
170
- TORCH_NIGHTLY_URL ,
170
+ TORCH_URL ,
171
171
],
172
172
check = True ,
173
173
)
You can’t perform that action at this time.
0 commit comments