ef240d5a61
install.sh's CUDA branch allows torch>=2.4,<2.12.0 and the cu12x/cu13x wheel indexes top out at torch 2.11.0, so a fresh NVIDIA install lands on torch 2.11. studio/backend/requirements/extras-no-deps.txt pinned torchcodec==0.10.0, which upstream pairs with torch 2.10 exactly, and torchcodec publishes no Requires-Dist: torch, so pip cannot notice. The guard that exists to catch this, _torchcodec_version_mismatch_hint, looks the torch minor up in _TORCH_TORCHCODEC_MINORS and returns None on a miss, and the table had no "2.11" row: it was silent on precisely the version where the mismatch now happens. Add the 2.11 row (mirrored into scripts/notebook_validator.py), add an audio-torch211 extra pinning the 0.11 line, and select torchcodec at install time from the venv's torch minor instead of a flat pin. A requirements file cannot branch on the installed torch version, so _select_torchcodec_spec follows the existing _select_torchao_spec pattern. It runs after the final torch repair, which can still move torch onto another minor. torchcodec 0.11 is the release upstream pairs with torch 2.11 exactly and the newest one published on the cu128 index install.sh resolves torch 2.11.0 from, since 0.12 dropped CUDA 12.8. From 0.12 torchcodec is ABI stable against torch >= 2.11, so that half of the matrix is open ended and cannot be written as a finite set of minors; both checkers now exempt torchcodec >= 0.12 on torch >= 2.11 rather than flagging a combination upstream supports, and torch 2.12+ installs take the open >=0.12.0 floor instead of the 2.11-only 0.11 line. Also bump the installer step total for the new step and keep torchcodec inside the security audit set now that it no longer comes from a requirements file.
25 lines
940 B
Text
25 lines
940 B
Text
# Audio extras (installed with --no-deps --no-cache-dir)
|
|
descript-audio-codec
|
|
descript-audiotools
|
|
julius
|
|
# torchcodec is NOT pinned here: its wheels are torch-ABI specific and a
|
|
# requirements file cannot branch on the installed torch minor (environment
|
|
# markers cannot see it). install_python_stack._select_torchcodec_spec picks
|
|
# the matching line right after this file is installed.
|
|
snac
|
|
|
|
# peft 0.19.0 causes export subprocess shutdown issues in Unsloth;
|
|
# installing with --no-deps to avoid pulling in torch>=0.11.0
|
|
peft==0.18.1
|
|
|
|
# TRL and related packages
|
|
trl==0.23.1
|
|
# executorch>=1.0.1 # 41.5 MB - no imports in unsloth/zoo/studio
|
|
torch-c-dlpack-ext
|
|
sentence_transformers==5.2.0
|
|
transformers==4.57.6
|
|
pytorch_tokenizers
|
|
kernels==0.12.1
|
|
# kernels<3.11 imports tomli as its tomllib fallback; --no-deps skips its own
|
|
# marker dep, so list it here (no-op on the 3.12/3.13 default installs).
|
|
tomli; python_version < "3.11"
|