diffusers/tools

1#!/usr/bin/env python32import torch3 4from diffusers import AutoencoderKL, StableDiffusionXLPipeline, StableDiffusionXLImg2ImgPipeline, KDPM2AncestralDiscreteScheduler5 6vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix",7    torch_dtype=torch.float168)9 10base = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0",11    vae=vae,12    torch_dtype=torch.float16,13    variant="fp16",14    use_safetensors=True15)16 17scheduler = KDPM2AncestralDiscreteScheduler.from_config(base.scheduler.config, use_karras_sigmas=True)18base.scheduler = scheduler19 20base.to("cuda")21 22def print_step(s, t, latents):23    print(s)24 25generator=torch.manual_seed(1111)26 27images = base(28    prompt="LOVE",29    num_inference_steps=10,30    generator=generator,31    callback=print_step32).images33