yoki22/kloe-lora

Instala las dependencias:

pip install -U diffusers transformers accelerate

Carga el LoRA con Diffusers para generar una imagen:

from diffusers import AutoPipelineForText2Image
import torch

pipeline = AutoPipelineForText2Image.from_pretrained(
    "stabilityai/stable-diffusion-xl-base-1.0",
    torch_dtype=torch.float16
).to("cuda")

pipeline.load_lora_weights(
    "yoki22/kloe-lora",
    weight_name="kloe_000001000.safetensors"
)

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

La ficha también muestra este ejemplo, etiquetado por Hugging Face como generación de texto a vídeo:

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

# Cambia "cuda" por "mps" en dispositivos Apple compatibles.
pipe = DiffusionPipeline.from_pretrained(
    "stabilityai/stable-diffusion-xl-base-1.0",
    torch_dtype=torch.bfloat16,
    device_map="cuda"
)
pipe.load_lora_weights("yoki22/kloe-lora")

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