Mark3228/baddie-4i-lora

Mark3228

Texto a video

Adaptador LoRA para el modelo base Wan-AI/Wan2.1-T2V-14B-Diffusers, orientado a la generación de vídeo a partir de texto. Fue entrenado con AI Toolkit de Ostris y sus pesos están disponibles en formato Safetensors.

Como usar

Instalación:
pip install -U diffusers transformers accelerate

Generación de vídeo con Diffusers:
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",
    torch_dtype=torch.bfloat16,
    device_map="cuda"
)
pipe.load_lora_weights("Mark3228/baddie-4i-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 incluye este ejemplo para cargar directamente el archivo Safetensors:
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(
    'Mark3228/baddie-4i-lora',
    weight_name='baddie-4i_000002750.safetensors'
)
image = pipeline('a beautiful landscape').images[0]
image.save("my_image.png")

Funcionalidades

Generación de vídeo a partir de texto mediante el modelo base Wan2.1-T2V de 14B parámetros
Adaptador LoRA compatible con la biblioteca Diffusers
Pesos disponibles en formato Safetensors
Compatible con herramientas como ComfyUI, AUTOMATIC1111, SD.Next e Invoke AI
No tiene palabras de activación definidas
Licencia CreativeML Open RAIL-M
No está desplegado actualmente en proveedores de inferencia

Casos de uso

Crear clips de vídeo a partir de descripciones textuales
Aplicar el estilo o ajuste aprendido por este LoRA sobre Wan2.1-T2V-14B
Integrar el adaptador en flujos locales de Diffusers o interfaces compatibles con LoRA
Experimentar con generación multimedia acelerada mediante GPU CUDA