[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 10:42:52 +00:00
commit 477757eac7
2 changed files with 5 additions and 2 deletions

View file

@ -430,7 +430,6 @@ def _resolve_base_precision(cfg, spec, device) -> str:
if mode == "mxfp8" and device == "cuda":
try:
import torch
blackwell = torch.cuda.get_device_capability() >= (10, 0)
except Exception: # noqa: BLE001 -- probe failure -> treat as unsupported, fail fast
blackwell = False

View file

@ -143,7 +143,11 @@ def test_family_train_infos_sdxl_supports_compile_without_precision_modes(monkey
# ── mxfp8 base precision (DiT dense speed mode) ───────────────────────────────
def _linear(in_features, out_features, bias = False):
def _linear(
in_features,
out_features,
bias = False,
):
import torch.nn as nn
return nn.Linear(in_features, out_features, bias = bias)