[diffusion] improve: improve code readability of DenoisingStage (#20003)

2026-03-06 23:23:44 +08:00
parent 54634b9a40
commit e8f2b80340
1 changed files with 2 additions and 2 deletions
@@ -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
        )