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