From 8cfe4820f6c5e19d5e41e4d3b1c283a643d40fdb Mon Sep 17 00:00:00 2001 From: Qianyi Zhao <49068354+Qing112@users.noreply.github.com> Date: Fri, 25 Oct 2024 03:23:01 -0500 Subject: [PATCH] Update sd_video.py --- diffsynth/pipelines/sd_video.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/diffsynth/pipelines/sd_video.py b/diffsynth/pipelines/sd_video.py index 99d65a4..e9be94a 100644 --- a/diffsynth/pipelines/sd_video.py +++ b/diffsynth/pipelines/sd_video.py @@ -185,7 +185,7 @@ class SDVideoPipeline(SDImagePipeline): if self.motion_modules is None: noise = self.generate_noise((1, 4, height//8, width//8), seed=seed, device="cpu", dtype=self.torch_dtype).repeat(num_frames, 1, 1, 1) else: - noise = self.generate_noise((1, 4, height//8, width//8), seed=seed, device="cpu", dtype=self.torch_dtype) + noise = self.generate_noise((num_frames, 4, height//8, width//8), seed=seed, device="cpu", dtype=self.torch_dtype) if input_frames is None or denoising_strength == 1.0: latents = noise else: