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