#!/usr/bin/env python3
from diffusers import DPMSolverMultistepScheduler, StableDiffusionXLPipeline, DPMSolverSDEScheduler

path = "hf-internal-testing/tiny-stable-diffusion-xl-pipe"

pipe = StableDiffusionXLPipeline.from_pretrained(path)
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config, algorithm_type="sde-dpmsolver++")

prompt = "An astronaut riding a green horse on Mars"
steps = 20

for i in range(2):
    width = 32 * (i + 1)
    height = 32 * (i + 1)
    image = pipe(prompt=prompt, width=width, height=height, num_inference_steps=steps).images[0]

pipe.scheduler = DPMSolverSDEScheduler.from_config(pipe.scheduler.config, algorithm_type="sde-dpmsolver++")

for i in range(2):
    width = 32 * (i + 1)
    height = 32 * (i + 1)
    image = pipe(prompt=prompt, width=width, height=height, num_inference_steps=steps).images[0]