[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-07-06 11:04:51 +00:00
commit cc6d7c96a6
2 changed files with 3 additions and 2 deletions

View file

@ -264,7 +264,6 @@ def bf16_unsupported_reason(resolved_family: str) -> Optional[str]:
return None
try:
import torch
if torch.cuda.is_available() and not torch.cuda.is_bf16_supported():
return (
"This trainer requires a bfloat16-capable GPU (Ampere or newer); this CUDA "

View file

@ -81,7 +81,9 @@ def test_base_precision_denies_fp8_for_corrupted_family():
# The deny is fp8-specific: int8 (per-token, unaffected) and the other dense modes stay
# allowed for the same Qwen base.
for mode in ("nf4", "bf16", "int8", "auto"):
norm = _cfg(base_model = _QWEN_DENSE, base_precision = mode, mixed_precision = "bf16").normalized()
norm = _cfg(
base_model = _QWEN_DENSE, base_precision = mode, mixed_precision = "bf16"
).normalized()
assert norm.resolved_family == "qwen-image"
assert norm.base_precision == mode