Views
No views yet
$ pip install diffusers transformers accelerate torch1from diffusers import DiffusionPipeline
2import torch
3from diffusers.utils import export_to_video
4
5pipe = DiffusionPipeline.from_pretrained(
6 "nachikethmurthy666/text-to-video-ms-1.7b",
7 torch_dtype=torch.float16,
8 variant="fp16"
9).to("cuda")
10
11pipe.enable_model_cpu_offload()
12
13prompt = "An astronaut riding a horse on Mars at sunset"
14video = pipe(prompt, num_inference_steps=25).frames[0]
15video_path = export_to_video(video)