From 4382836b97dd8c9ec87530d1842c6e7f7f08f115 Mon Sep 17 00:00:00 2001 From: h1t Date: Mon, 9 Sep 2024 15:26:06 +0000 Subject: [PATCH] Fix dynCFG timesteps --- sat/sgm/modules/diffusionmodules/sampling.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/sat/sgm/modules/diffusionmodules/sampling.py b/sat/sgm/modules/diffusionmodules/sampling.py index f0f1830..318d0f5 100644 --- a/sat/sgm/modules/diffusionmodules/sampling.py +++ b/sat/sgm/modules/diffusionmodules/sampling.py @@ -515,7 +515,10 @@ class VideoDDIMSampler(BaseDiffusionSampler): ).to(torch.float32) if isinstance(self.guider, DynamicCFG): denoised = self.guider( - denoised, (1 - alpha_cumprod_sqrt**2) ** 0.5, step_index=self.num_steps - timestep, scale=scale + denoised, + (1 - alpha_cumprod_sqrt**2) ** 0.5, + step_index=self.discretization.num_steps - timestep, + scale=scale, ) else: denoised = self.guider(denoised, (1 - alpha_cumprod_sqrt**2) ** 0.5, scale=scale)