jude1903/AJWWan22-lora

jude1903

Texto a video

Adaptador LoRA para el modelo base Wan2.2-T2V-A14B, entrenado con AI Toolkit de Ostris. Está orientado a generar contenido visual de una mujer mediante la palabra de activación «ajw woman» y ofrece pesos en formato Safetensors.

Como usar

Instalación de las bibliotecas necesarias:
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" 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("jude1903/AJWWan22-lora")
prompt = "ajw woman"
output = pipe(prompt=prompt).frames[0]
export_to_video(output, "output.mp4")

Generación de imagen utilizando el archivo de pesos LoRA 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(
    'jude1903/AJWWan22-lora',
    weight_name='AJWWan22_low_noise.safetensors'
)
image = pipeline('ajw woman').images[0]
image.save("my_image.png")

Funcionalidades

Adaptador LoRA basado en ai-toolkit/Wan2.2-T2V-A14B-Diffusers-bf16.
Compatible con generación de texto a vídeo mediante Diffusers.
Incluye pesos Safetensors para flujos de generación de imágenes.
Utiliza «ajw woman» como palabra de activación.
Puede emplearse con Diffusers, ComfyUI, AUTOMATIC1111, SD.Next e Invoke AI.
Distribuido bajo la licencia CreativeML Open RAIL-M.

Casos de uso

Generar vídeos condicionados por el texto de activación «ajw woman».
Generar imágenes con los pesos AJWWan22_low_noise.safetensors.
Integrar el estilo o sujeto aprendido por el LoRA en flujos locales basados en Wan2.2.
Usar el adaptador en interfaces compatibles con LoRA y Safetensors.