From 9ac6bbe67de7583c16b3ad8978b5b2fbf464e2c9 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 16 Jul 2026 09:29:00 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- .../backend/core/inference/diffusion_transformer_quant.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/studio/backend/core/inference/diffusion_transformer_quant.py b/studio/backend/core/inference/diffusion_transformer_quant.py index 3de18ece1c..da1f37da62 100644 --- a/studio/backend/core/inference/diffusion_transformer_quant.py +++ b/studio/backend/core/inference/diffusion_transformer_quant.py @@ -439,10 +439,12 @@ def _make_quant_config(scheme: str, fast_accum: Optional[bool] = None) -> Any: # (torchao >= 0.13); older versions keep today's behaviour via the signature check. import inspect from torchao.quantization import PerRow + fp8_kwargs: dict = {"granularity": PerRow()} - if "activation_value_lb" in inspect.signature( - Float8DynamicActivationFloat8WeightConfig - ).parameters: + if ( + "activation_value_lb" + in inspect.signature(Float8DynamicActivationFloat8WeightConfig).parameters + ): fp8_kwargs["activation_value_lb"] = 1e-12 try: from torchao.float8 import Float8MMConfig