xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2025-12-16 08:24:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix onediff

Browse Source
This commit is contained in:
Jukka Seppänen 2024-09-03 19:26:49 +03:00
parent c8c55256b2
commit 3d536025e0
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pipeline_cogvideox.py
View File

@ -617,6 +617,7 @@ class CogVideoXPipeline(DiffusionPipeline):
**extra_step_kwargs, **extra_step_kwargs,
return_dict=False, return_dict=False,
) )
latents = latents.to(prompt_embeds.dtype)
# start diff diff # start diff diff
if i < len(timesteps) - 1 and self.original_mask is not None: if i < len(timesteps) - 1 and self.original_mask is not None:
noise_timestep = timesteps[i + 1] noise_timestep = timesteps[i + 1]