From a06aa7c3754af02e997a7a424bc67d5429b8e72b Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Sat, 6 Jan 2024 11:47:50 +0530 Subject: [PATCH] fix: vae type --- .../stable_video_diffusion/pipeline_stable_video_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py b/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py index 4cc0905a9910..fa96f41cd81f 100644 --- a/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py +++ b/src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py @@ -77,7 +77,7 @@ class StableVideoDiffusionPipeline(DiffusionPipeline): implemented for all pipelines (downloading, saving, running on a particular device, etc.). Args: - vae ([`AutoencoderKL`]): + vae ([`AutoencoderKLTemporalDecoder`]): Variational Auto-Encoder (VAE) model to encode and decode images to and from latent representations. image_encoder ([`~transformers.CLIPVisionModelWithProjection`]): Frozen CLIP image-encoder ([laion/CLIP-ViT-H-14-laion2B-s32B-b79K](https://huggingface.co/laion/CLIP-ViT-H-14-laion2B-s32B-b79K)).