File size: 265 Bytes
a6ddc0e
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
from diffusers import DiffusionPipeline
import torch

pipe = DiffusionPipeline.from_pretrained(
    "damo-vilab/text-to-video-ms-1.7b",
    torch_dtype=torch.float16
).to("cuda")

prompt = "a futuristic city at night, neon lights"
video_frames = pipe(prompt).frames