Skip to content

Commit 8b6fae4

Browse files
authored
[SVD] fix: vae type (#6475)
fix: vae type
1 parent aa1797e commit 8b6fae4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ class StableVideoDiffusionPipeline(DiffusionPipeline):
7777
implemented for all pipelines (downloading, saving, running on a particular device, etc.).
7878
7979
Args:
80-
vae ([`AutoencoderKL`]):
80+
vae ([`AutoencoderKLTemporalDecoder`]):
8181
Variational Auto-Encoder (VAE) model to encode and decode images to and from latent representations.
8282
image_encoder ([`~transformers.CLIPVisionModelWithProjection`]):
8383
Frozen CLIP image-encoder ([laion/CLIP-ViT-H-14-laion2B-s32B-b79K](https://huggingface.co/laion/CLIP-ViT-H-14-laion2B-s32B-b79K)).

0 commit comments

Comments
 (0)