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

Raise ImportError on stable torch/torchvision mismatch (#4065)

Browse source 
* Raise ImportError for stable torchvision mismatches

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

for more information, see https://pre-commit.ci

* Remove torchvision compatibility tests from PR scope

---------

Co-authored-by: Daniel Hanchen <danielhanchen@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Daniel Han 2026-02-15 19:14:19 -08:00 • committed by GitHub
commit ec80fd3f66
1 changed files with 4 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

14
unsloth/import_fixes.py
View file

@ -668,7 +668,6 @@ def torchvision_compatibility_check():
# Try known table first, then fall back to formula for forward compatibility
required = TORCH_TORCHVISION_COMPAT.get((torch_major, torch_minor))
is_in_known_table = required is not None
if required is None:
required = _infer_required_torchvision(torch_major, torch_minor)
@ -705,15 +704,10 @@ def torchvision_compatibility_check():
t in torchvision_version_raw for t in _pre_tags
)
# Downgrade to warning for custom/source/pre-release builds or formula-predicted
if is_custom or is_prerelease or not is_in_known_table:
reason = (
"custom/source build"
if is_custom
else "pre-release build"
if is_prerelease
else "newer torch version"
)
# Only downgrade to warning for custom/source or prerelease builds.
# Stable mismatches should fail fast to prevent runtime operator errors.
if is_custom or is_prerelease:
reason = "custom/source build" if is_custom else "pre-release build"
logger.warning(
f"{message}\n"
f"Detected a {reason}. "