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

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-05-24 15:24:35 +00:00
commit f116f78b1d
2 changed files with 4 additions and 1 deletions

View file

@ -50,7 +50,9 @@ def check_torch() -> tuple[int, int]:
if cap[0] < 7 or (cap[0] == 7 and cap[1] < 5):
sys.exit(f"FAIL: pre-Turing GPU {name} is not supported by this image")
if cap[0] < 8:
print(f"NOTE: {name} is Turing (sm_{cap[0]}{cap[1]}) -- bf16 unavailable, fp16 fallback.")
print(
f"NOTE: {name} is Turing (sm_{cap[0]}{cap[1]}) -- bf16 unavailable, fp16 fallback."
)
return cap

View file

@ -166,6 +166,7 @@ if os.environ.get("UNSLOTH_FORCE_SINGLE_COMPILE_WORKER", "0") == "1":
os.environ["TORCHINDUCTOR_COMPILE_THREADS"] = "1"
try:
from unsloth_zoo.temporary_patches import common as _zoo_common
_zoo_common.determine_compile_threads = lambda: 1
except Exception:
pass