From c5b99a1f93df08690d4bd0abdcf70f9587c973f9 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Sat, 4 Jul 2026 01:30:42 +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_dit_trainer.py | 8 +++++++- .../backend/core/training/diffusion_training_service.py | 4 +++- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/studio/backend/core/training/diffusion_dit_trainer.py b/studio/backend/core/training/diffusion_dit_trainer.py index b4aa754402..2a03c9ad0c 100644 --- a/studio/backend/core/training/diffusion_dit_trainer.py +++ b/studio/backend/core/training/diffusion_dit_trainer.py @@ -280,7 +280,13 @@ def _apply_fp8_training(transformer, on_event) -> bool: def _pick_auto_precision( - prequant, device, free_gb, dense_gb, capability, has_fp8, has_torchao = True + prequant, + device, + free_gb, + dense_gb, + capability, + has_fp8, + has_torchao = True, ) -> str: """Pure policy for base_precision="auto": nf4 for a prequant base or no CUDA; else the fastest dense mode whose weights + headroom (activations, optimizer, cache) fit the diff --git a/studio/backend/core/training/diffusion_training_service.py b/studio/backend/core/training/diffusion_training_service.py index b33b061f1d..67f64f14ad 100644 --- a/studio/backend/core/training/diffusion_training_service.py +++ b/studio/backend/core/training/diffusion_training_service.py @@ -450,7 +450,9 @@ class DiffusionTrainingService: if "learning_rate" in ev else s["learning_rate"] ) - grad_norm = _finite_or_none(ev["grad_norm"]) if "grad_norm" in ev else s["grad_norm"] + grad_norm = ( + _finite_or_none(ev["grad_norm"]) if "grad_norm" in ev else s["grad_norm"] + ) s.update( status = "running", step = ev.get("step", s["step"]),