From fefa2187f077185b8c13ee8eae28333ab9b5ad8f 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:06 +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 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/studio/backend/core/training/diffusion_dit_trainer.py b/studio/backend/core/training/diffusion_dit_trainer.py
index 1cbdd93c53..d6288ce48c 100644
--- a/studio/backend/core/training/diffusion_dit_trainer.py
+++ b/studio/backend/core/training/diffusion_dit_trainer.py
@@ -279,7 +279,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