JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-12-02 10:32:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix dynCFG timesteps

Browse Source
This commit is contained in:
h1t 2024-09-09 15:26:06 +00:00
parent f47370a601
commit 4382836b97
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
sat/sgm/modules/diffusionmodules/sampling.py
View File

@ -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)