探索者7622from diffusers import StableVideoDiffusionPipeline
import torch
# 加载模型
pipe = StableVideoDiffusionPipeline.from_pretrained(
"stabilityai/stable-video-diffusion-img2vid-xt",
torch_dtype=torch.float16,
variant="fp16"
)
pipe.to("cuda")
# 使用一张霸王龙图像作为起始帧 (需要提前准备)
init_image = load_image("t_rex_hunting_pose.jpg")
# 生成视频
frames = pipe(
init_image,
height=1024,
width=576,
num_frames=24*18, # 18秒视频
fps=24,
motion_bucket_id=180, # 更高的运动程度
noise_aug_strength=0.1
).frames[0]
# 保存视频
export_to_video(frames, "t_rex_hunt.mp4", fps=24)
内容由AI生成