diffusers/tools

1#!/usr/bin/env python32from diffusers import DPMSolverMultistepScheduler, StableDiffusionXLPipeline, DPMSolverSDEScheduler3 4path = "hf-internal-testing/tiny-stable-diffusion-xl-pipe"5 6pipe = StableDiffusionXLPipeline.from_pretrained(path)7pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, algorithm_type="sde-dpmsolver++")8 9prompt = "An astronaut riding a green horse on Mars"10steps = 2011 12for i in range(2):13    width = 32 * (i + 1)14    height = 32 * (i + 1)15    image = pipe(prompt=prompt, width=width, height=height, num_inference_steps=steps).images[0]16 17pipe.scheduler = DPMSolverSDEScheduler.from_config(pipe.scheduler.config, algorithm_type="sde-dpmsolver++")18 19for i in range(2):20    width = 32 * (i + 1)21    height = 32 * (i + 1)22    image = pipe(prompt=prompt, width=width, height=height, num_inference_steps=steps).images[0]23