Update pipeline_cogvideox.py

This commit is contained in:
kijai 2024-11-10 19:31:37 +02:00
parent ed167ecfff
commit 184097e78e

View File

@ -865,7 +865,6 @@ class CogVideoXPipeline(VideoSysPipeline, CogVideoXLoraLoaderMixin):
video_flow_features=video_flow_features if (tora is not None and tora["start_percent"] <= current_step_percentage <= tora["end_percent"]) else None, video_flow_features=video_flow_features if (tora is not None and tora["start_percent"] <= current_step_percentage <= tora["end_percent"]) else None,
)[0] )[0]
noise_pred = noise_pred.float() noise_pred = noise_pred.float()
print(self._guidance_scale[i])
if isinstance(self.scheduler, CogVideoXDPMScheduler): if isinstance(self.scheduler, CogVideoXDPMScheduler):
self._guidance_scale[i] = 1 + guidance_scale[i] * ( self._guidance_scale[i] = 1 + guidance_scale[i] * (
(1 - math.cos(math.pi * ((num_inference_steps - t.item()) / num_inference_steps) ** 5.0)) / 2 (1 - math.cos(math.pi * ((num_inference_steps - t.item()) / num_inference_steps) ** 5.0)) / 2