From 6cfa4c9c2f70be5f9c5bbe24d9ceaeaa2a2c9a80 Mon Sep 17 00:00:00 2001 From: WenhaoZhang <42087078+niehen6174@users.noreply.github.com> Date: Fri, 15 May 2026 13:22:24 +0800 Subject: [PATCH] [diffusion] fix: respect dit_precision config instead of hardcoded bfloat16 in DenoisingStagge (#24988) Co-authored-by: niehen6174 --- .../multimodal_gen/runtime/pipelines_core/stages/denoising.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sglang/multimodal_gen/runtime/pipelines_core/stages/denoising.py b/python/sglang/multimodal_gen/runtime/pipelines_core/stages/denoising.py index ef56c3b8a..fa99647c8 100644 --- a/python/sglang/multimodal_gen/runtime/pipelines_core/stages/denoising.py +++ b/python/sglang/multimodal_gen/runtime/pipelines_core/stages/denoising.py @@ -618,7 +618,7 @@ class DenoisingStage(PipelineStage, RolloutDenoisingMixin): ) # Setup precision and autocast settings - target_dtype = torch.bfloat16 + target_dtype = PRECISION_TO_TYPE[server_args.pipeline_config.dit_precision] autocast_enabled = ( target_dtype != torch.float32 ) and not server_args.disable_autocast