update

model.py

@@ -2,14 +2,20 @@ from __future__ import annotations
 
 
 from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
+from diffusers import DPMSolverMultistepScheduler
 import torch
 import PIL.Image
 import numpy as np
 
+device = "cpu"
+
 class Model:
     def __init__(self):
         modelID = "runwayml/stable-diffusion-v1-5"
-        pipeline = StableDiffusionPipeline.from_pretrained(modelID, torch_dtype=torch.float16)
+        pipe = StableDiffusionPipeline.from_pretrained(modelID, torch_dtype=torch.float16)
+        pipe = pipe.to(device)
+        pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+        
         #self.pipe = StableDiffusionPipeline.from_pretrained(modelID)
         #prompt = "a photo of an astronaut riding a horse on mars"
         #n_prompt = "deformed, disfigured"
@@ -22,13 +28,13 @@ class Model:
                 num_steps:int = 20,
                 ):
         seed = np.random.randint(0, np.iinfo(np.int64).max)
-        generator = torch.Generator().manual_seed(seed)
+        generator = torch.Generator(device).manual_seed(seed)
         return self.pipe(prompt=prompt,
                          negative_prompt=negative_prompt,
                          guidance_scale=guidance_scale,
                          num_images_per_prompt=num_images,
                          num_inference_steps=num_steps,
-                         generator=generator).images        
+                         generator=generator).images