Skip to content

Commit 4e7b0cb

Browse files
authored
[docs] fix: animatediff docs (#6339)
fix: animatediff docs
1 parent 35b81ff commit 4e7b0cb

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

src/diffusers/pipelines/animatediff/pipeline_animatediff.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,14 @@
3333
LMSDiscreteScheduler,
3434
PNDMScheduler,
3535
)
36-
from ...utils import USE_PEFT_BACKEND, BaseOutput, logging, scale_lora_layers, unscale_lora_layers
36+
from ...utils import (
37+
USE_PEFT_BACKEND,
38+
BaseOutput,
39+
logging,
40+
replace_example_docstring,
41+
scale_lora_layers,
42+
unscale_lora_layers,
43+
)
3744
from ...utils.torch_utils import randn_tensor
3845
from ..pipeline_utils import DiffusionPipeline
3946

@@ -47,7 +54,7 @@
4754
>>> from diffusers import MotionAdapter, AnimateDiffPipeline, DDIMScheduler
4855
>>> from diffusers.utils import export_to_gif
4956
50-
>>> adapter = MotionAdapter.from_pretrained("diffusers/motion-adapter")
57+
>>> adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-2")
5158
>>> pipe = AnimateDiffPipeline.from_pretrained("frankjoshua/toonyou_beta6", motion_adapter=adapter)
5259
>>> pipe.scheduler = DDIMScheduler(beta_schedule="linear", steps_offset=1, clip_sample=False)
5360
>>> output = pipe(prompt="A corgi walking in the park")
@@ -533,6 +540,7 @@ def prepare_latents(
533540
return latents
534541

535542
@torch.no_grad()
543+
@replace_example_docstring(EXAMPLE_DOC_STRING)
536544
def __call__(
537545
self,
538546
prompt: Union[str, List[str]] = None,

0 commit comments

Comments
 (0)