From f116f78b1d761122c03d0eeee6dc042f0d9fc893 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Sun, 24 May 2026 15:24:35 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- docker/smoke_test.py | 4 +++- unsloth/_gpu_init.py | 1 + 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/docker/smoke_test.py b/docker/smoke_test.py index 6e8fd60340..eeeeb49f13 100644 --- a/docker/smoke_test.py +++ b/docker/smoke_test.py @@ -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 diff --git a/unsloth/_gpu_init.py b/unsloth/_gpu_init.py index 5f9abfb026..221e08ba18 100644 --- a/unsloth/_gpu_init.py +++ b/unsloth/_gpu_init.py @@ -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