How to use fal/flux-klein-9b-virtual-tryon-lora with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline from diffusers.utils import load_image # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("fill-in-base-model", dtype=torch.bfloat16, device_map="cuda") pipe.load_lora_weights("fal/flux-klein-9b-virtual-tryon-lora") prompt = "Turn this cat into a dog" 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]
From my tests, adding LoRA does slightly improve color consistency, but the results are still not that different from those without LoRA. Is this normal? Also, do you have a sample workflow?
· Sign up or log in to comment
