Skip to content

Commit dfccb1c

Browse files
committed
Remove torchcodec from install_requirements.py
Summary: torchcodec is used when only torchaudio users use torch.save and torch.load. It is an optional requirement. Only moshi test uses torch.save and torch.load, and we should make this to install across the board. Currently, it is blocking linux aarch64 wheel building (#13946) because during smoke testing it is trying to install torchcodec on linux aarch64 machine and it is failing
1 parent 2805a81 commit dfccb1c

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

examples/models/moshi/mimi/install_requirements.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
set -x
99

1010
conda install "ffmpeg<8"
11+
pip install torchcodec==0.7.0.dev20250906 --extra-index-url https://download.pytorch.org/whl/nightly/cpu
1112
pip install moshi==0.2.4
1213
pip install bitsandbytes soundfile
1314
# Run llama2/install requirements for torchao deps

install_requirements.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -155,9 +155,6 @@ def install_optional_example_requirements(use_pytorch_nightly):
155155
else "torchvision"
156156
),
157157
f"torchaudio==2.8.0.{NIGHTLY_VERSION}" if use_pytorch_nightly else "torchaudio",
158-
# We probably don't need explicit torchcodec, see comment here
159-
# https://github.com/pytorch/audio/pull/4039#issuecomment-3274648824
160-
f"torchcodec==0.7.0.{NIGHTLY_VERSION}" if use_pytorch_nightly else "torchcodec",
161158
]
162159
# Then install domain libraries
163160
subprocess.run(

0 commit comments

Comments
 (0)