From e8f2b80340e2d07dea89431ede77689eef1c303d Mon Sep 17 00:00:00 2001 From: JohnHerry Date: Fri, 6 Mar 2026 23:23:44 +0800 Subject: [PATCH] [diffusion] improve: improve code readability of DenoisingStage (#20003) --- .../multimodal_gen/runtime/pipelines_core/stages/denoising.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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 48646bb1a..f06343174 100644 --- a/python/sglang/multimodal_gen/runtime/pipelines_core/stages/denoising.py +++ b/python/sglang/multimodal_gen/runtime/pipelines_core/stages/denoising.py @@ -343,9 +343,9 @@ class DenoisingStage(PipelineStage): torch.full( (batch_size,), guidance_val, - dtype=torch.float32, + dtype=target_dtype, device=device, - ).to(target_dtype) + ) * 1000.0 )