How to use Photoroom/FLUX.2-klein-4b-fp8-diffusers 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("Photoroom/FLUX.2-klein-4b-fp8-diffusers", dtype=torch.bfloat16, device_map="cuda") 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]
Dear Developers,
Could you please provide a quantized version of the 9B model? Alternatively, if that is not available, would it be possible to share the quantization script with us?
Thank you so much for your time and help!
· Sign up or log in to comment
