Spaces:
Running
Running
File size: 480 Bytes
27435c3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from diffusers import DiffusionPipeline
import torch
device = "cuda" if torch.cuda.is _available() else "cpu"
pipeline=DiffusionPipeline.from_pretrained("stabilityai/stable-video-diffusion-img2vid-xt")
pipeline = pipeline.to(devices)
prompt = "A beautiful sunset over the ocean"
video = pipeline(prompt,num_interface_steps=50).videos
video path = "generated_video.mp4"with open(video_path, "wb") as f:
f.write(video[0])
print(f"video generated and saved as {video_path}") |