fwqww/darkfantasywan-lora

Instalación y generación de vídeo con Diffusers:

pip install -U diffusers transformers accelerate

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

# Cambia a "mps" en dispositivos Apple
pipe = DiffusionPipeline.from_pretrained(
    "Wan-AI/Wan2.1-T2V-14B-Diffusers",
    dtype=torch.bfloat16,
    device_map="cuda"
)
pipe.load_lora_weights("fwqww/darkfantasywan-lora")

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

La ficha también proporciona este ejemplo para cargar un archivo de pesos concreto y generar una imagen, aunque su uso de una canalización de texto a imagen parece inconsistente con el modelo base de texto a vídeo:

from diffusers import AutoPipelineForText2Image
import torch

pipeline = AutoPipelineForText2Image.from_pretrained(
    'Wan-AI/Wan2.1-T2V-14B-Diffusers',
    torch_dtype=torch.bfloat16
).to('cuda')
pipeline.load_lora_weights(
    'fwqww/darkfantasywan-lora',
    weight_name='darkfantasywan_000002750.safetensors'
)
image = pipeline('darkfantasy').images[0]
image.save("my_image.png")