Skip to content

Commit 81d3247

Browse files
committed
update docstring example
1 parent 3e12970 commit 81d3247

2 files changed

Lines changed: 4 additions & 3 deletions

File tree

src/diffusers/pipelines/hunyuan_image/pipeline_hunyuanimage.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@
4747
>>> from diffusers import HunyuanImagePipeline
4848
4949
>>> pipe = HunyuanImagePipeline.from_pretrained(
50-
... "hunyuanvideo-community/HunyuanVideo", torch_dtype=torch.bfloat16
50+
... "hunyuanvideo-community/HunyuanImage-2.1-Diffusers", torch_dtype=torch.bfloat16
5151
... )
5252
>>> pipe.to("cuda")
5353
>>> prompt = "A cat holding a sign that says hello world"

src/diffusers/pipelines/hunyuan_image/pipeline_hunyuanimage_refiner.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,13 +46,14 @@
4646
>>> from diffusers import HunyuanImageRefinerPipeline
4747
4848
>>> pipe = HunyuanImageRefinerPipeline.from_pretrained(
49-
... "hunyuanvideo-community/HunyuanImageRefiner", torch_dtype=torch.bfloat16
49+
... "hunyuanvideo-community/HunyuanImage-2.1-Refiner-Diffusers", torch_dtype=torch.bfloat16
5050
... )
5151
>>> pipe.to("cuda")
5252
>>> prompt = "A cat holding a sign that says hello world"
53+
>>> image = load_image("path/to/image.png")
5354
>>> # Depending on the variant being used, the pipeline call will slightly vary.
5455
>>> # Refer to the pipeline documentation for more details.
55-
>>> image = pipe(prompt, negative_prompt="", num_inference_steps=50).images[0]
56+
>>> image = pipe(prompt, image=image, num_inference_steps=4).images[0]
5657
>>> image.save("hunyuanimage.png")
5758
```
5859
"""

0 commit comments

Comments
 (0)