Update pipeline_cogvideox.py

This commit is contained in:
kijai 2024-08-27 21:51:07 +03:00
parent a8251b3b93
commit 44a8305fcb

View File

@ -569,7 +569,7 @@ class CogVideoXPipeline(DiffusionPipeline):
if do_classifier_free_guidance:
noise_pred_uncond, noise_pred_text = noise_pred.chunk(2)
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
latents, old_pred_original_sample = self.scheduler.step(