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 )