xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2025-12-16 00:14: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

5
pipeline_cogvideox.py
View File

@ -172,7 +172,7 @@ class CogVideoXPipeline(DiffusionPipeline):
) )
noise = randn_tensor(shape, generator=generator, device=device, dtype=self.vae.dtype) noise = randn_tensor(shape, generator=generator, device=device, dtype=self.vae.dtype)
if latents is None: if latents is None:
latents = noise latents = noise
else: else:
latents = latents.to(device) latents = latents.to(device)
timesteps, num_inference_steps = self.get_timesteps(num_inference_steps, denoise_strength, device) timesteps, num_inference_steps = self.get_timesteps(num_inference_steps, denoise_strength, device)
@ -538,7 +538,7 @@ class CogVideoXPipeline(DiffusionPipeline):
noise_pred = noise_pred_uncond + self._guidance_scale * (noise_pred_text - noise_pred_uncond) noise_pred = noise_pred_uncond + self._guidance_scale * (noise_pred_text - noise_pred_uncond)
# compute the previous noisy sample x_t -> x_t-1 # compute the previous noisy sample x_t -> x_t-1
latents_tile = self.scheduler.step(noise_pred, t, latents_tile.to(self.vae.dtype), **extra_step_kwargs, return_dict=False)[0] latents_tile = self.scheduler.step(noise_pred, t, latents_tile.to(self.vae.dtype), **extra_step_kwargs, return_dict=False)[0]
latents_all_list.append(latents_tile) latents_all_list.append(latents_tile)
# ========================================== # ==========================================
@ -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]