File size: 510 Bytes
903f240 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
import torch
from diffusers import DiffusionPipeline
from diffusers.utils import load_image, export_to_video
pipe = DiffusionPipeline.from_pretrained("Skywork/Matrix-Game-2.0", torch_dtype=torch.float16)
pipe.to("cuda")
prompt = "A man with short gray hair plays a red electric guitar."
image = load_image(
"https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/guitar-man.png"
)
output = pipe(image=image, prompt=prompt).frames[0]
export_to_video(output, "output.mp4") |