Skip to content

Commit 6f90bc1

Browse files
a-r-r-o-wsayakpaul
authored andcommitted
[docs] fix pia example (#9015)
fix pia example docstring
1 parent ceeaf1d commit 6f90bc1

File tree

1 file changed

+7
-8
lines changed

1 file changed

+7
-8
lines changed

src/diffusers/pipelines/pia/pipeline_pia.py

+7-8
Original file line numberDiff line numberDiff line change
@@ -54,22 +54,21 @@
5454
Examples:
5555
```py
5656
>>> import torch
57-
>>> from diffusers import (
58-
... EulerDiscreteScheduler,
59-
... MotionAdapter,
60-
... PIAPipeline,
61-
... )
57+
>>> from diffusers import EulerDiscreteScheduler, MotionAdapter, PIAPipeline
6258
>>> from diffusers.utils import export_to_gif, load_image
6359
64-
>>> adapter = MotionAdapter.from_pretrained("../checkpoints/pia-diffusers")
65-
>>> pipe = PIAPipeline.from_pretrained("SG161222/Realistic_Vision_V6.0_B1_noVAE", motion_adapter=adapter)
60+
>>> adapter = MotionAdapter.from_pretrained("openmmlab/PIA-condition-adapter")
61+
>>> pipe = PIAPipeline.from_pretrained(
62+
... "SG161222/Realistic_Vision_V6.0_B1_noVAE", motion_adapter=adapter, torch_dtype=torch.float16
63+
... )
64+
6665
>>> pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
6766
>>> image = load_image(
6867
... "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/pix2pix/cat_6.png?download=true"
6968
... )
7069
>>> image = image.resize((512, 512))
7170
>>> prompt = "cat in a hat"
72-
>>> negative_prompt = "wrong white balance, dark, sketches,worst quality,low quality, deformed, distorted, disfigured, bad eyes, wrong lips,weird mouth, bad teeth, mutated hands and fingers, bad anatomy,wrong anatomy, amputation, extra limb, missing limb, floating,limbs, disconnected limbs, mutation, ugly, disgusting, bad_pictures, negative_hand-neg"
71+
>>> negative_prompt = "wrong white balance, dark, sketches, worst quality, low quality, deformed, distorted"
7372
>>> generator = torch.Generator("cpu").manual_seed(0)
7473
>>> output = pipe(image=image, prompt=prompt, negative_prompt=negative_prompt, generator=generator)
7574
>>> frames = output.frames[0]

0 commit comments

Comments
 (0)