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

Update svd_video.py

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

2
diffsynth/pipelines/svd_video.py
View File

@@ -49,7 +49,7 @@ class SVDVideoPipeline(BasePipeline):
return image_emb return image_emb
def encode_image_with_vae(self, image, noise_aug_strength, seed): def encode_image_with_vae(self, image, noise_aug_strength, seed=None):
image = self.preprocess_image(image).to(device=self.device, dtype=self.torch_dtype) image = self.preprocess_image(image).to(device=self.device, dtype=self.torch_dtype)
noise = self.generate_noise(image.shape, seed=seed, device=self.device, dtype=self.torch_dtype) noise = self.generate_noise(image.shape, seed=seed, device=self.device, dtype=self.torch_dtype)
image = image + noise_aug_strength * noise image = image + noise_aug_strength * noise