FLUX.1 Krea [dev]

Puede usarse como reemplazo directo de FLUX.1 [dev] en sistemas compatibles. Está disponible en ComfyUI mediante descarga de pesos .safetensors, y en diffusers con FluxPipeline.

Primero, instala o actualiza diffusers:

pip install -U diffusers

Ejemplo con diffusers:

import torch
from diffusers import FluxPipeline

pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-Krea-dev", torch_dtype=torch.bfloat16)
pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU VRAM

prompt = "A frog holding a sign that says hello world"
image = pipe(
    prompt,
    height=1024,
    width=1024,
    guidance_scale=4.5,
).images[0]
image.save("flux-krea-dev.png")

También se indica una implementación de referencia de FLUX.1 [dev] en su repositorio dedicado de GitHub y documentación adicional en diffusers.