FLUX.1-Kontext-dev-LoRA-Pixel-Style

import torch
from diffusers import FluxKontextPipeline
from diffusers.utils import load_image

pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16)
pipe.load_lora_weights("Shakker-Labs/FLUX.1-Kontext-dev-LoRA-Pixel-Style", weight_name="FLUX-kontext-lora-pixel-style.safetensors")
pipe.fuse_lora(lora_scale=1.0)
pipe.to("cuda")

input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")
prompt = "Convert to a pixel art style"

image = pipe(
    image=input_image,
    prompt=prompt, 
    num_inference_steps=24, 
    guidance_scale=2.5,
).images[0]
image.save(f"example.png")