diffusers/tools

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