diffusers/tools

1#!/usr/bin/env python32from diffusers import (3    SASolverScheduler,4    StableDiffusionPipeline,5    KDPM2DiscreteScheduler,6    StableDiffusionImg2ImgPipeline,7    HeunDiscreteScheduler,8    KDPM2AncestralDiscreteScheduler,9    DDIMScheduler,10)11import time12import torch13import sys14from pathlib import Path15 16# path = sys.argv[1]17path = "runwayml/stable-diffusion-v1-5"18device = "mps"19 20pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16)21pipe.scheduler = SASolverScheduler.from_config(pipe.scheduler.config)22pipe.to(device)23 24prompt = "A lion in galaxies, spirals, nebulae, stars, smoke, iridescent, intricate detail, octane render, 8k"25 26start_time = time.time()27generator = torch.Generator(device=device).manual_seed(0)28images = pipe(29    prompt=prompt, generator=generator, num_images_per_prompt=1, num_inference_steps=3030).images31print(time.time() - start_time)32 33images[0].show()34