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