Ashmotv/animat3d_style_wan-lora

Instala las dependencias y carga el LoRA sobre el modelo base Wan2.2:

pip install -U diffusers transformers accelerate

import torch
from diffusers import DiffusionPipeline
from diffusers.utils import export_to_video

# Usa "mps" en dispositivos Apple compatibles
pipe = DiffusionPipeline.from_pretrained(
    "ai-toolkit/Wan2.2-T2V-A14B-Diffusers-bf16",
    dtype=torch.bfloat16,
    device_map="cuda"
)
pipe.load_lora_weights("Ashmotv/animat3d_style_wan-lora")

prompt = "animat3d_style"
output = pipe(prompt=prompt).frames[0]
export_to_video(output, "output.mp4")

También puede cargarse un archivo de pesos específico:

from diffusers import AutoPipelineForText2Image
import torch

pipeline = AutoPipelineForText2Image.from_pretrained(
    "ai-toolkit/Wan2.2-T2V-A14B-Diffusers-bf16",
    torch_dtype=torch.bfloat16
).to("cuda")

pipeline.load_lora_weights(
    "Ashmotv/animat3d_style_wan-lora",
    weight_name="animat3d_style_wan_low_noise.safetensors"
)

image = pipeline("animat3d_style").images[0]
image.save("my_image.png")