Genera imágenes con IA y Stable Diffusion: guía práctica en Python

Mira, colega, si has llegado hasta aquí es porque estás harto de ver tutoriales que te cuentan la historia de la IA desde 1950 y no te enseñan ni una línea de código útil. Vamos al grano: hoy toca ensuciarse las manos con Stable Diffusion y Python. Nada de teoría vacía, solo código que funciona y prompts que puedes copiar y pegar.

Lo que necesitas antes de escribir código

Primero, instala las dependencias. No hay excusa, es un solo comando:

pip install diffusers transformers accelerate safetensors torch

Si tienes una GPU con al menos 6GB de VRAM, genial. Si no, también funciona con CPU, pero tendrás tiempo de sobra para hacer un café mientras esperas.

El código mínimo que funciona

Abre tu editor y pega esto. Es el esqueleto básico que usarás para el 90% de tus pruebas:

from diffusers import StableDiffusionPipeline

import torch

Cargar el modelo (la primera vez tarda un poco)

pipe = StableDiffusionPipeline.from_pretrained(

"runwayml/stable-diffusion-v1-5",

torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32

)

if torch.cuda.is_available():

pipe = pipe.to("cuda")

Generar la imagen

prompt = "un robot pintando un mural futurista en una calle de Madrid, estilo cyberpunk, luz neón"

imagen = pipe(prompt, num_inference_steps=30).images[0]

Guardarla

imagen.save("mi_primera_imagen_ia.png")

print("Imagen generada sin drama, colega.")

Eso es todo. No hay magia, solo código. Ahora, lo interesante: cómo sacarle partido.

Prompts que funcionan de verdad

Un buen prompt marca la diferencia entre una imagen mediocre y una que parece sacada de una película. Aquí tienes algunos probados:

Prompt para estilo cinematográfico:

prompt = "retrato de una mujer con gafas de sol reflejando un atardecer, iluminación cinematográfica, profundidad de campo, estilo película de ciencia ficción"

Prompt para arte conceptual:

prompt = "ciudad flotante en las nubes, arquitectura imposible, arte conceptual, detalles intrincados, iluminación dramática, estilo matte painting"

Prompt para producto:

prompt = "zapatillas deportivas futuristas flotando en un pedestal, fondo minimalista rojo, iluminación de estudio profesional, fotografía de producto"

Trucos para que tus imágenes no parezcan de 2022

Mira, el modelo base está bien, pero si quieres resultados que impresionen, usa estos dos trucos:

1. Ajusta el CFG (Classifier-Free Guidance): El valor por defecto es 7.5, pero subirlo a 10-12 te da imágenes más fieles al prompt, aunque con menos libertad creativa.

imagen = pipe(prompt, num_inference_steps=40, guidance_scale=10).images[0]

2. Usa un sampler diferente: El "Euler a" o "DPM++ 2M Karras" suelen dar mejores resultados que el que viene por defecto:

from diffusers import DPMSolverMultistepScheduler

pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)

Integrar con una API para producción

¿Quieres montar un endpoint para tu app? Aquí tienes lo más básico para no morir en el intento:

from fastapi import FastAPI

from pydantic import BaseModel

app = FastAPI()

class PromptRequest(BaseModel):

prompt: str

steps: int = 30

@app.post("/generar")

async def generar(data: PromptRequest):

imagen = pipe(data.prompt, num_inference_steps=data.steps).images[0]

imagen.save("temp.png")

return {"mensaje": "Listo", "archivo": "temp.png"}

Errores comunes que te van a pasar

• "CUDA out of memory": Reduce el tamaño de la imagen con pipe(height=512, width=512) o usa .half() en el modelo.

• Resultados raros: Añade "high quality, detailed, sharp focus" al final de tus prompts.

• Tiempo de espera eterno: Baja los pasos a 20-25. La calidad no cae tanto como crees.

Y una cosa más: Stable Diffusion no es perfecto. A veces las manos salen mal, las letras parecen garabatos y ciertos conceptos abstractos se le escapan. Pero con práctica, consigues resultados que te sorprenden.

¿Tienes algún truco que haya funcionado bien para ti? Compártelo, colega, que esto del arte digital con IA se disfruta más en comunidad. Y tú, ¿qué herramienta usas para generar imágenes con IA en tus proyectos?