From e9db36a7ca0d61d0e4152dc62b9c1a97fd92a159 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 8 Jul 2026 05:02:09 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- studio/backend/core/training/diffusion_train_common.py | 1 + studio/backend/tests/test_diffusion_base_precision.py | 8 ++++++-- studio/backend/tests/test_diffusion_dit_trainer.py | 4 +++- 3 files changed, 10 insertions(+), 3 deletions(-) diff --git a/studio/backend/core/training/diffusion_train_common.py b/studio/backend/core/training/diffusion_train_common.py index d62cbd1707..f67a6fd0f6 100644 --- a/studio/backend/core/training/diffusion_train_common.py +++ b/studio/backend/core/training/diffusion_train_common.py @@ -304,6 +304,7 @@ def native_bf16_supported() -> bool: guard so all three stay in sync.""" try: import torch + if not torch.cuda.is_available(): return False is_rocm = bool(getattr(getattr(torch, "version", None), "hip", None)) diff --git a/studio/backend/tests/test_diffusion_base_precision.py b/studio/backend/tests/test_diffusion_base_precision.py index 61ddcad959..630f22cf37 100644 --- a/studio/backend/tests/test_diffusion_base_precision.py +++ b/studio/backend/tests/test_diffusion_base_precision.py @@ -146,7 +146,9 @@ def test_bf16_unsupported_reason(monkeypatch): # report is_bf16_supported() True, so the gate is native compute capability (major >= 8), not # is_bf16_supported() -- otherwise the emulation case would slip through and evict-then-fail. monkeypatch.setattr(torch.cuda, "is_available", lambda: True) - monkeypatch.setattr(torch.cuda, "is_bf16_supported", lambda *a, **k: True) # emulation reports True + monkeypatch.setattr( + torch.cuda, "is_bf16_supported", lambda *a, **k: True + ) # emulation reports True monkeypatch.setattr(torch.cuda, "get_device_capability", lambda *a, **k: (7, 5)) # Turing assert "bfloat16" in (bf16_unsupported_reason("flux.1") or "") @@ -168,7 +170,9 @@ def test_native_bf16_supported_gates_on_capability(monkeypatch): from core.training.diffusion_train_common import native_bf16_supported monkeypatch.setattr(torch.cuda, "is_available", lambda: True) - monkeypatch.setattr(torch.cuda, "is_bf16_supported", lambda *a, **k: True) # emulation reports True + monkeypatch.setattr( + torch.cuda, "is_bf16_supported", lambda *a, **k: True + ) # emulation reports True monkeypatch.setattr(torch.cuda, "get_device_capability", lambda *a, **k: (7, 5)) # Turing assert native_bf16_supported() is False monkeypatch.setattr(torch.cuda, "get_device_capability", lambda *a, **k: (8, 0)) # Ampere diff --git a/studio/backend/tests/test_diffusion_dit_trainer.py b/studio/backend/tests/test_diffusion_dit_trainer.py index 8ba1940c89..b1fb7d21bb 100644 --- a/studio/backend/tests/test_diffusion_dit_trainer.py +++ b/studio/backend/tests/test_diffusion_dit_trainer.py @@ -311,7 +311,9 @@ def test_train_precision_modes_pre_ampere_is_nf4_only(monkeypatch): import torch monkeypatch.setattr(torch.cuda, "is_available", lambda: True) - monkeypatch.setattr(torch.cuda, "is_bf16_supported", lambda *a, **k: True) # emulation reports True + monkeypatch.setattr( + torch.cuda, "is_bf16_supported", lambda *a, **k: True + ) # emulation reports True monkeypatch.setattr(torch.cuda, "get_device_capability", lambda *a, **k: (7, 5)) # Turing modes, recommended = train_precision_modes() assert modes == ["nf4"]