# Demo: FLUX.3 Image Klein – Bildgenerierung mit diffusers
import torch
from diffusers import DiffusionPipeline
from diffusers.utils import load_image

REPO = "black-forest-labs/flux-3-image-klein"
device = "cuda" if torch.cuda.is_available() else ("mps" if torch.backends.mps.is_available() else "cpu")
dtype = torch.float16 if device != "cpu" else torch.float32
pipe = DiffusionPipeline.from_pretrained(REPO, torch_dtype=dtype, trust_remote_code=True)
if device == "cuda":
    pipe.enable_model_cpu_offload()   # spart VRAM
else:
    pipe.to(device)
prompt = "Ein fotorealistischer roter Fuchs im verschneiten Wald, goldene Stunde"
init = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")
image = pipe(prompt=prompt, image=init).images[0]
image.save("output.png")
print("Bild gespeichert: output.png")