Skip to content

Commit e36a027

Browse files
authored
[RELEASE] Remove nightly pin from install_requirements.py (#5402)
1 parent eecf74f commit e36a027

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

install_requirements.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -97,12 +97,12 @@ def python_is_compatible():
9797
NIGHTLY_VERSION = "dev20240901"
9898

9999
# 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"
101101

102102
# pip packages needed by exir.
103103
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.
106106
"typing-extensions",
107107
]
108108

@@ -121,7 +121,7 @@ def python_is_compatible():
121121
# TODO: Make each example publish its own requirements.txt
122122
EXAMPLES_REQUIREMENTS = [
123123
"timm==1.0.7",
124-
f"torchaudio==2.5.0.{NIGHTLY_VERSION}",
124+
"torchaudio==2.5.0",
125125
"torchsr==1.0.4",
126126
"transformers==4.42.4",
127127
]
@@ -140,7 +140,7 @@ def python_is_compatible():
140140
"install",
141141
*REQUIREMENTS_TO_INSTALL,
142142
"--extra-index-url",
143-
TORCH_NIGHTLY_URL,
143+
TORCH_URL,
144144
],
145145
check=True,
146146
)
@@ -167,7 +167,7 @@ def python_is_compatible():
167167
"--no-build-isolation",
168168
"-v",
169169
"--extra-index-url",
170-
TORCH_NIGHTLY_URL,
170+
TORCH_URL,
171171
],
172172
check=True,
173173
)

0 commit comments

Comments
 (0)