ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-05-06 21:34:34 +00:00
commit ec40e9f207
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
studio/install_python_stack.py
View file

@ -65,7 +65,12 @@ _ROCM_WINDOWS_WHEEL_BASE = (
).rstrip("/")
# Maps (major, minor) → (release_folder, torch_ver, torchvision_ver, torchaudio_ver)
_ROCM_WINDOWS_RELEASES: dict[tuple[int, int], tuple[str, str, str, str]] = {
(7, 2): ("rocm-rel-7.2.1", "2.9.1+rocm7.2.1", "0.24.1+rocm7.2.1", "2.9.1+rocm7.2.1"),
(7, 2): (
"rocm-rel-7.2.1",
"2.9.1+rocm7.2.1",
"0.24.1+rocm7.2.1",
"2.9.1+rocm7.2.1",
),
}
# bitsandbytes continuous-release_main wheels with the ROCm 4-bit GEMV fix
@ -295,9 +300,7 @@ def _ensure_rocm_torch() -> None:
entry = next(
(
v
for (maj, mn), v in sorted(
_ROCM_WINDOWS_RELEASES.items(), reverse = True
)
for (maj, mn), v in sorted(_ROCM_WINDOWS_RELEASES.items(), reverse = True)
if ver >= (maj, mn)
),
None,