strangerzonehf/Flux-Super-Portrait-LoRA

Instalación y uso con Diffusers:

pip install -U diffusers transformers accelerate

import torch
from diffusers import DiffusionPipeline

# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained(
    "black-forest-labs/FLUX.1-dev",
    dtype=torch.bfloat16,
    device_map="cuda"
)
pipe.load_lora_weights("strangerzonehf/Flux-Super-Portrait-LoRA")

prompt = "Super Portrait, A close-up portrait of a young man with dark brown eyes and dark brown eyebrows. He is wearing a green and yellow striped polo shirt with a black collar. His earring is adorned with a silver earring. The backdrop is a light blue color."
image = pipe(prompt).images[0]

Configuración alternativa indicada en la página:

import torch
from pipelines import DiffusionPipeline

base_model = "black-forest-labs/FLUX.1-dev"
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)

lora_repo = "strangerzonehf/Flux-Super-Portrait-LoRA"
trigger_word = "Super Portrait"
pipe.load_lora_weights(lora_repo)

device = torch.device("cuda")
pipe.to(device)

Para activar el estilo del modelo, incluir Super Portrait al inicio del prompt.