prithivMLmods/Digital-Chaos-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/Digital-Chaos-Flux-LoRA")

prompt = "Digital Chaos, An abstract painting of a hamburger. The hamburger is made up of a variety of colors, including red, orange, yellow, green, blue, and purple. The bun is adorned with sesame seeds, adding a pop of color to the painting. The background is a vibrant shade of red, purple, and blue, creating a striking contrast with the burger. The burger is positioned in the center of the frame, with the bun facing the viewer."
image = pipe(prompt).images[0]

Configuración alternativa indicada en la tarjeta:

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

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