From 955fcff828705e685e1ad119ebd117940f480481 Mon Sep 17 00:00:00 2001 From: Lancer Date: Tue, 12 May 2026 10:08:28 +0800 Subject: [PATCH] [Chore] explicit .float() conversion in Helios's optimized_scale function (#3529) Signed-off-by: Lancer --- vllm_omni/diffusion/models/helios/pipeline_helios.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/vllm_omni/diffusion/models/helios/pipeline_helios.py b/vllm_omni/diffusion/models/helios/pipeline_helios.py index df709a515..c3a47ebaa 100644 --- a/vllm_omni/diffusion/models/helios/pipeline_helios.py +++ b/vllm_omni/diffusion/models/helios/pipeline_helios.py @@ -53,6 +53,8 @@ def calculate_shift( def optimized_scale(positive_flat, negative_flat): + positive_flat = positive_flat.float() + negative_flat = negative_flat.float() dot_product = torch.sum(positive_flat * negative_flat, dim=1, keepdim=True) squared_norm = torch.sum(negative_flat**2, dim=1, keepdim=True) + 1e-8 st_star = dot_product / squared_norm