From 9a87e3f723b3c659985526cae835ddc0c6e79f46 Mon Sep 17 00:00:00 2001 From: Mr-Neutr0n <64578610+Mr-Neutr0n@users.noreply.github.com> Date: Wed, 11 Feb 2026 19:47:56 +0530 Subject: [PATCH] fix: use max_scale for trapezoid plateau instead of 1.0 --- sgm/modules/diffusionmodules/guiders.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sgm/modules/diffusionmodules/guiders.py b/sgm/modules/diffusionmodules/guiders.py index db2cede0d..411f07477 100644 --- a/sgm/modules/diffusionmodules/guiders.py +++ b/sgm/modules/diffusionmodules/guiders.py @@ -147,7 +147,7 @@ def __init__( self.scale = torch.cat( [ rise_steps, - torch.ones(num_frames - 2 * int(num_frames * edge_perc)), + torch.full((num_frames - 2 * int(num_frames * edge_perc),), max_scale), fall_steps, ] ).unsqueeze(0)