fix i2v resolution

This commit is contained in:
Artiprocher
2025-06-24 10:56:52 +08:00
parent 3b3e1e4d44
commit 93194f44e8
4 changed files with 14 additions and 8 deletions

View File

@@ -19,12 +19,13 @@ state_dict = load_state_dict("models/train/Wan2.1-I2V-14B-720P_full/epoch-1.safe
pipe.dit.load_state_dict(state_dict)
pipe.enable_vram_management()
input_image = VideoData("data/example_video_dataset/video1.mp4", height=480, width=832)[0]
input_image = VideoData("data/example_video_dataset/video1.mp4", height=720, width=1280)[0]
video = pipe(
prompt="from sunset to night, a small town, light, house, river",
negative_prompt="色调艳丽过曝静态细节模糊不清字幕风格作品画作画面静止整体发灰最差质量低质量JPEG压缩残留丑陋的残缺的多余的手指画得不好的手部画得不好的脸部畸形的毁容的形态畸形的肢体手指融合静止不动的画面杂乱的背景三条腿背景人很多倒着走",
input_image=input_image,
height=720, width=1280, num_frames=49,
seed=1, tiled=True
)
save_video(video, "video_Wan2.1-I2V-14B-720P.mp4", fps=15, quality=5)