Skip to content

Commit f6a07e5

Browse files
committed
update example
1 parent 0a5a820 commit f6a07e5

File tree

1 file changed

+18
-9
lines changed

1 file changed

+18
-9
lines changed

src/diffusers/pipelines/hunyuan_video/pipeline_hunyuan_video_image2video.py

Lines changed: 18 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -51,23 +51,32 @@
5151
Examples:
5252
```python
5353
>>> import torch
54-
>>> from diffusers import HunyuanVideoPipeline, HunyuanVideoTransformer3DModel
55-
>>> from diffusers.utils import export_to_video
54+
>>> from diffusers import HunyuanVideoImageToVideoPipeline, HunyuanVideoTransformer3DModel
55+
>>> from diffusers.utils import load_image, export_to_video
5656
57-
>>> model_id = "hunyuanvideo-community/HunyuanVideo"
57+
>>> model_id = "hunyuanvideo-community/HunyuanVideo-I2V"
5858
>>> transformer = HunyuanVideoTransformer3DModel.from_pretrained(
5959
... model_id, subfolder="transformer", torch_dtype=torch.bfloat16
6060
... )
61-
>>> pipe = HunyuanVideoPipeline.from_pretrained(model_id, transformer=transformer, torch_dtype=torch.float16)
61+
>>> pipe = HunyuanVideoImageToVideoPipeline.from_pretrained(
62+
... model_id, transformer=transformer, torch_dtype=torch.float16
63+
... )
6264
>>> pipe.vae.enable_tiling()
6365
>>> pipe.to("cuda")
6466
67+
>>> prompt = "A man with short gray hair plays a red electric guitar."
68+
>>> image = load_image(
69+
... "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/guitar-man.png"
70+
... )
71+
6572
>>> output = pipe(
66-
... prompt="A cat walks on the grass, realistic",
67-
... height=320,
68-
... width=512,
69-
... num_frames=61,
70-
... num_inference_steps=30,
73+
... image=image,
74+
... height=720,
75+
... width=1280,
76+
... num_frames=129,
77+
... prompt=prompt,
78+
... true_cfg_scale=1.0,
79+
... guidance_scale=1.0,
7180
... ).frames[0]
7281
>>> export_to_video(output, "output.mp4", fps=15)
7382
```

0 commit comments

Comments
 (0)