Skip to content

GitLab

Unverified Commit 67b3fe0a authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

Fix resize issue in SVD pipeline with VideoProcessor (#8229) 



update
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent baab0656
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py
View file @ 67b3fe0a
...@@ -178,7 +178,7 @@ class StableVideoDiffusionPipeline(DiffusionPipeline): ...@@ -178,7 +178,7 @@ class StableVideoDiffusionPipeline(DiffusionPipeline):
feature_extractor=feature_extractor, feature_extractor=feature_extractor,
) )
self.vae_scale_factor = 2 ** (len(self.vae.config.block_out_channels) - 1) self.vae_scale_factor = 2 ** (len(self.vae.config.block_out_channels) - 1)
self.video_processor = VideoProcessor(do_resize=False, vae_scale_factor=self.vae_scale_factor) self.video_processor = VideoProcessor(do_resize=True, vae_scale_factor=self.vae_scale_factor)
def _encode_image( def _encode_image(
self, self,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment