from diffusers import StableDiffusionXLPipeline

import torch

 

pipe = StableDiffusionXLPipeline.from_pretrained(

    "stabilityai/sdxl-turbo",

    torch_dtype=torch.float16,

    variant="fp16"

).to("mps")  # mps = Apple Metal GPU

 

pipe.set_progress_bar_config(disable=False)

 

prompt = "a mystical forest with glowing trees, cinematic lighting, 4k"

image = pipe(prompt=prompt, guidance_scale=0.0, num_inference_steps=4).images[0]

image.save("turbo_output.png")