Mademon/centa_wan-lora

Mademon

Texto a video

Adaptador LoRA para generación de vídeo a partir de texto, entrenado con AI Toolkit sobre el modelo base Wan2.2-T2V-A14B en formato Diffusers BF16. Sus pesos están disponibles en formato Safetensors y no requiere palabras de activación.

Como usar

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" para dispositivos Apple
pipe = DiffusionPipeline.from_pretrained(
    "ai-toolkit/Wan2.2-T2V-A14B-Diffusers-bf16",
    torch_dtype=torch.bfloat16,
    device_map="cuda"
)
pipe.load_lora_weights("Mademon/centa_wan-lora")

prompt = "A man with short gray hair plays a red electric guitar."
output = pipe(prompt=prompt).frames[0]
export_to_video(output, "output.mp4")

La página también proporciona este ejemplo para cargar explícitamente los pesos de bajo ruido:
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(
    'Mademon/centa_wan-lora',
    weight_name='centa_wan_low_noise.safetensors'
)
image = pipeline('a beautiful landscape').images[0]
image.save("my_image.png")

Funcionalidades

Generación de vídeo a partir de texto
Adaptador LoRA basado en Wan2.2-T2V-A14B
Entrenado con AI Toolkit de Ostris
Pesos disponibles en formato Safetensors
Compatible con la biblioteca Diffusers
Utilizable con ComfyUI, AUTOMATIC1111, SD.Next e Invoke AI
No requiere palabras de activación
Licencia CreativeML Open RAIL-M

Casos de uso

Crear vídeos a partir de descripciones textuales
Aplicar el estilo o ajuste aprendido por el LoRA al modelo Wan2.2-T2V-A14B
Integrar generación audiovisual personalizada en flujos locales de Diffusers
Usar los pesos LoRA en interfaces compatibles como ComfyUI o Invoke AI