diff --git a/pipeline_cogvideox.py b/pipeline_cogvideox.py index 877a9f7..b96e2fe 100644 --- a/pipeline_cogvideox.py +++ b/pipeline_cogvideox.py @@ -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(