[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
a9b8d68b57
commit
f116f78b1d
2 changed files with 4 additions and 1 deletions
|
|
@ -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
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue