diff --git a/src/diffusers/pipelines/mochi/pipeline_mochi.py b/src/diffusers/pipelines/mochi/pipeline_mochi.py index 9d460f7693..c56002753b 100644 --- a/src/diffusers/pipelines/mochi/pipeline_mochi.py +++ b/src/diffusers/pipelines/mochi/pipeline_mochi.py @@ -48,6 +48,7 @@ EXAMPLE_DOC_STRING = """ ```py >>> import torch >>> from diffusers import MochiPipeline + >>> from diffusers.utils import export_to_video >>> pipe = MochiPipeline.from_pretrained("genmo/mochi-1-preview", torch_dtype=torch.bfloat16) >>> pipe.to("cuda")