# Demo: MOSS-Image-SFT – Bildgenerierung mit diffusers
import torch
from diffusers import DiffusionPipeline
from diffusers.utils import load_image

REPO = "OpenMOSS-Team/MOSS-Image-SFT"
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"
image = pipe(prompt).images[0]
image.save("output.png")
print("Bild gespeichert: output.png")