Spaces:
Runtime error
Runtime error
File size: 426 Bytes
adced09 dfa4f73 74ff47f dfa4f73 74ff47f dfa4f73 |
1 2 3 4 5 6 7 8 9 10 11 12 |
import torch
from diffusers import WanPipeline
from diffusers.utils import export_to_video
pipe = WanPipeline.from_pretrained(
"Wan-AI/Wan2.1-T2V-1.3B-Diffusers", torch_dtype=torch.bfloat16
).to("cuda")
pipe.load_lora_weights("finetrainers/Wan2.1-T2V-1.3B-3dgs-v0", adapter_name="wan-lora")
pipe.set_adapters(["wan-lora"], [0.5])
video = pipe("<my-awesome-prompt>").frames[0]
export_to_video(video, "output.mp4", fps=24) |