基础模型:

• 腾讯/HunyuanVideo-I2V 库名称: diffusers 管道标签: 图像转视频

非官方社区分叉版本，提供Diffusers格式权重的tencent/HunyuanVideo-I2V模型。

使用Diffusers

HunyuanVideo-I2V可直接通过Diffusers使用。请安装最新版本的Diffusers。

import torch
from diffusers import HunyuanVideoImageToVideoPipeline, HunyuanVideoTransformer3DModel
from diffusers.utils import load_image, export_to_video

# 可用检查点: "hunyuanvideo-community/HunyuanVideo-I2V" 和 "hunyuanvideo-community/HunyuanVideo-I2V-33ch"
model_id = "hunyuanvideo-community/HunyuanVideo-I2V"
transformer = HunyuanVideoTransformer3DModel.from_pretrained(
    model_id, subfolder="transformer", torch_dtype=torch.bfloat16
)
pipe = HunyuanVideoImageToVideoPipeline.from_pretrained(
    model_id, transformer=transformer, torch_dtype=torch.float16
)
pipe.vae.enable_tiling()
pipe.to("cuda")

prompt = "一位灰白色短发的男子正在弹奏红色电吉他。"
image = load_image(
    "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/guitar-man.png"
)

output = pipe(image=image, prompt=prompt).frames[0]
export_to_video(output, "output.mp4", fps=15)

更多信息请参阅文档。