import torch
from diffusers import DiffusionPipeline
from diffusers.utils import load_image
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", dtype=torch.bfloat16, device_map="cuda")
pipe.load_lora_weights("gokaygokay/Light-Fix-Kontext-Dev-LoRA")
prompt = "Fuse this image into background"
input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")
image = pipe(image=input_image, prompt=prompt).images[0]Light Fix Kontext Dev LoRA
- Prompt
- Fuse this image into background
- Prompt
- Fuse this image into background
- Prompt
- Fuse this image into background
Model description
Trigger phrase
Fuse this image into background
Download model
Weights for this model are available in Safetensors format.
Training at fal.ai
Training was done using fal.ai/models/fal-ai/flux-kontext-trainer/playground.
- Downloads last month
- 208
Model tree for gokaygokay/Light-Fix-Kontext-Dev-LoRA
Base model
black-forest-labs/FLUX.1-Kontext-dev