Skip to content

Commit 804b018

Browse files
committed
Add torch into core requirements and torchaudio and torchvision into requirements-examples.txt
1 parent a977bc4 commit 804b018

File tree

3 files changed

+2
-22
lines changed

3 files changed

+2
-22
lines changed

install_requirements.py

Lines changed: 0 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -143,26 +143,6 @@ def install_optional_example_requirements(use_pytorch_nightly):
143143
check=True,
144144
)
145145

146-
print("Installing torch domain libraries")
147-
DOMAIN_LIBRARIES = [
148-
("torchvision==0.23.0" if use_pytorch_nightly else "torchvision"),
149-
"torchaudio==2.8.0" if use_pytorch_nightly else "torchaudio",
150-
]
151-
# Then install domain libraries
152-
subprocess.run(
153-
[
154-
sys.executable,
155-
"-m",
156-
"pip",
157-
"install",
158-
*DOMAIN_LIBRARIES,
159-
"--extra-index-url",
160-
TORCH_URL,
161-
],
162-
check=True,
163-
)
164-
165-
166146
# Prebuilt binaries for Intel-based macOS are no longer available on PyPI; users must compile from source.
167147
# PyTorch stopped building macOS x86_64 binaries since version 2.3.0 (January 2024).
168148
def is_intel_mac_os():

pyproject.toml

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,8 +68,6 @@ dependencies=[
6868
"ruamel.yaml",
6969
"sympy",
7070
"torch>=2.8.0,<2.9.0",
71-
"torchaudio>=2.8.0,<2.9.0",
72-
"torchvision>=0.23.0,<0.24.0",
7371
"torchao==0.12.0",
7472
"tabulate",
7573
# See also third-party/TARGETS for buck's typing-extensions version.

requirements-examples.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,3 +5,5 @@ timm == 1.0.7
55
torchsr == 1.0.4
66
torchtune >= 0.6.1
77
transformers ==4.47.1
8+
torchaudio>=2.8.0,<2.9.0
9+
torchvision>=0.23.0,<0.24.0

0 commit comments

Comments
 (0)