Views
No views yet
1pip install diffusers transformers accelerate imageio[ffmpeg]
2pip install git+ssh://git@github.com/baaivision/NOVA.git1import torch
2from diffnext.pipelines import NOVAPipeline
3from diffnext.utils import export_to_image, export_to_video
4
5model_id = "BAAI/nova-d48w1024-osp480"
6model_args = {"torch_dtype": torch.float16, "trust_remote_code": True}
7pipe = NOVAPipeline.from_pretrained(model_id, **model_args)
8pipe = pipe.to("cuda")
9
10prompt = "Many spotted jellyfish pulsating under water."
11
12image = pipe(prompt, max_latent_length=1).frames[0, 0]
13export_to_image(image, "jellyfish.jpg")
14
15video = pipe(prompt, max_latent_length=9).frames[0]
16export_to_video(video, "jellyfish.mp4", fps=12)
17
18# Increase AR and diffusion steps for better video quality.
19video = pipe(
20 prompt,
21 max_latent_length=9,
22 num_inference_steps=128, # default: 64
23 num_diffusion_steps=100, # default: 25
24).frames[0]
25export_to_video(video, "jellyfish_v2.mp4", fps=12)