prithivMLmods/Logo-Design-Flux-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("prithivMLmods/Logo-Design-Flux-LoRA")

prompt = "Logo Design, A close-up of a bottle cap with the word \"HUNTER\" carved into the center of the cap. The cap is made up of a silver metal frame with ridges along the edges. The background is a dark gray color. The word is carved into a wood-like pattern. There are tall pine trees in the middle of the frame. There is a mountain range in the background."
image = pipe(prompt).images[0]

Configuración alternativa indicada en la ficha:

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 = "prithivMLmods/Logo-Design-Flux-LoRA"
trigger_word = "Logo Design"
pipe.load_lora_weights(lora_repo)

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