sdxl-lora-juguete-v1

import torch
from diffusers import DiffusionPipeline, AutoencoderKL

# 0. Verificación de seguridad (GPU Check)
if not torch.cuda.is_available():
    raise RuntimeError("❌ ERROR: No se detecta GPU. En Colab, vete a 'Entorno de ejecución' > 'Cambiar tipo' > 'T4 GPU'.")

# 1. Cargar el modelo base
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
pipe = DiffusionPipeline.from_pretrained(
    "stabilityai/stable-diffusion-xl-base-1.0",
    vae=vae,
    torch_dtype=torch.float16,
    variant="fp16",
    use_safetensors=True
)
pipe.to("cuda")

# 2. Cargar el LoRA
pipe.load_lora_weights("Juanpeg1729/sdxl-lora-juguete-v1")

# 3. Generar imagen
print("Generando...")
prompt = "photo of sks bear toy in a cyberpunk city, neon lights, 8k"
image = pipe(prompt, num_inference_steps=30).images[0]

# 4. Guardar y Mostrar
image.save("resultado.png") 
image