prithivMLmods/Ton618-Only-Stickers-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/Ton618-Only-Stickers-Flux-LoRA")

prompt = "Only Sticker, An animated illustration of a sloth in a mug with the words \"Sloth Fuel\" written on it. The sloth is holding the mug with its left hand and its right hand resting on the mug. The mug is filled with coffee and there is a brown liquid dripping from the bottom of the cup. There is a white background behind the sloth with a smoke coming out of it. There are three coffee beans on the bottom left corner of the image."
image = pipe(prompt).images[0]

Configuración alternativa indicada en la tarjeta del modelo:

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/Ton618-Only-Stickers-Flux-LoRA"
trigger_word = "Only Sticker"
pipe.load_lora_weights(lora_repo)

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

Para activar correctamente el estilo, los prompts deberían empezar con Only Sticker.