breaking-bad-flux

markury

Texto a imagen

Este es un modelo LoRA derivado de black-forest-labs/FLUX.1-dev. El prompt principal usado durante la validación fue: una fotografía de Walter White y Jesse Pinkman. El codificador de texto no fue entrenado. Puede reutilizar el codificador de texto del modelo base para la inferencia.

Como usar

import torch
from diffusers import DiffusionPipeline

model_id = 'black-forest-labs/FLUX.1-dev'
adapter_id = 'markury/breaking-bad-flux'
pipeline = DiffusionPipeline.from_pretrained(model_id)
pipeline.load_lora_weights(adapter_id)

prompt = "a photograph of Walter White and Jesse Pinkman"

pipeline.to('cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu')
image = pipeline(
prompt=prompt,
num_inference_steps=28,
generator=torch.Generator(device='cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu').manual_seed(1641421826),
width=1024,
height=1024,
guidance_scale=3.5,
).images[0]
image.save("output.png", format="PNG")

Funcionalidades

Modelo derivado de LoRA
Usa el prompt de validación: una fotografía de Walter White y Jesse Pinkman
Establece la configuración de validación con CFG de 3.5, rescale de 0.0, 28 pasos, y resolución de 1024x1024
No entrenó el codificador de texto, reutiliza el del modelo base
Configuración de entrenamiento con 176 épocas, 3000 pasos y tasa de aprendizaje de 0.0001
Utiliza el optimizador AdamW con precisión BF16 pura

Casos de uso

Generación de imágenes de personajes de Walter White y Jesse Pinkman
Uso en proyectos de arte digital relacionados con la serie Breaking Bad