theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-24 18:28:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update sd_video.py

Browse Source
This commit is contained in:
Qianyi Zhao
2024-10-25 03:23:01 -05:00
committed by GitHub
parent c8021d4224
commit 8cfe4820f6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
diffsynth/pipelines/sd_video.py
View File

@@ -185,7 +185,7 @@ class SDVideoPipeline(SDImagePipeline):
if self.motion_modules is None: 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) 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: 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: if input_frames is None or denoising_strength == 1.0:
latents = noise latents = noise
else: else: