[feat] add support for Kandinsky5 I2V

This commit is contained in:
leffff
2026-07-07 14:35:48 -07:00
committed by SolitaryThinker
parent 6de9d6ee87
commit ed1fbee9b0
7 changed files with 178 additions and 37 deletions
@@ -1,9 +1,10 @@
from fastvideo import VideoGenerator
OUTPUT_PATH = "video_samples_kandinsky5_2B_t2v"
OUTPUT_PATH = "video_samples_kandinsky5_2B_i2v"
IMAGE_PATH = "assets/girl.png"
def main():
generator = VideoGenerator.from_pretrained(
"kandinskylab/Kandinsky-5.0-I2V-Lite-sft-5s-Diffusers",
@@ -22,8 +23,16 @@ def main():
"brings it to her lips to take a sip. After drinking, she places the cup back on "
"the table and looks out the window, enjoying the peaceful atmosphere."
)
_ = generator.generate_video(prompt, output_path=OUTPUT_PATH, save_video=True,height=512, width=768, num_frames=121)
_ = generator.generate_video(
prompt,
image_path=IMAGE_PATH,
output_path=OUTPUT_PATH,
save_video=True,
height=512,
width=768,
num_frames=121,
)
if __name__ == "__main__":
main()
main()