TurkeyVideoai / python
1c1's picture
Create Python
a6ddc0e verified
raw
history blame contribute delete
265 Bytes
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