Views
No views yet
python gen.py --model-path FastDM/Wan2.2-I2V-A14B-Merge-Lightning-V1.0-Diffusers --architecture wan-i2v --guidance-scale 1.0 --height 480 --width 832 --steps 4 --use-fp8 --output-path ./output.mp4 --num-frames 81 --fps 16 --task i2v --prompts [PROMPTS] --image-path [PATH/TO/IMAGE]1import torch
2import numpy as np
3from diffusers import WanImageToVideoPipeline
4from diffusers.utils import export_to_video, load_image
5
6model_id = "FastDM/Wan2.2-I2V-A14B-Merge-Lightning-V1.0-Diffusers"
7dtype = torch.bfloat16
8device = "cuda"
9
10pipe = WanImageToVideoPipeline.from_pretrained(model_id, torch_dtype=dtype)
11pipe.to(device)
12
13
14image = load_image(
15 "https://huggingface.co/datasets/YiYiXu/testing-images/resolve/main/wan_i2v_input.JPG"
16)
17max_area = 480 * 832
18aspect_ratio = image.height / image.width
19mod_value = pipe.vae_scale_factor_spatial * pipe.transformer.config.patch_size[1]
20height = round(np.sqrt(max_area * aspect_ratio)) // mod_value * mod_value
21width = round(np.sqrt(max_area / aspect_ratio)) // mod_value * mod_value
22image = image.resize((width, height))
23prompt = "Summer beach vacation style, a white cat wearing sunglasses sits on a surfboard. The fluffy-furred feline gazes directly at the camera with a relaxed expression. Blurred beach scenery forms the background featuring crystal-clear waters, distant green hills, and a blue sky dotted with white clouds. The cat assumes a naturally relaxed posture, as if savoring the sea breeze and warm sunlight. A close-up shot highlights the feline's intricate details and the refreshing atmosphere of the seaside."
24
25negative_prompt = "色调艳丽,过曝,静态,细节模糊不清,字幕,风格,作品,画作,画面,静止,整体发灰,最差质量,低质量,JPEG压缩残留,丑陋的,残缺的,多余的手指,画得不好的手部,画得不好的脸部,畸形的,毁容的,形态畸形的肢体,手指融合,静止不动的画面,杂乱的背景,三条腿,背景人很多,倒着走"
26generator = torch.Generator(device=device).manual_seed(0)
27output = pipe(
28 image=image,
29 prompt=prompt,
30 negative_prompt=negative_prompt,
31 height=height,
32 width=width,
33 num_frames=81,
34 guidance_scale=1.0,
35 num_inference_steps=4,
36 generator=generator,
37).frames[0]
38export_to_video(output, "i2v_output.mp4", fps=16)
39