karimbenharrak commited on
Commit
27eaf2b
1 Parent(s): 43a4c7b

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -1,6 +1,6 @@
1
  from typing import Dict, List, Any
2
  import torch
3
- from diffusers import DPMSolverMultistepScheduler, EulerDiscreteScheduler, DDIMScheduler, StableDiffusionInpaintPipeline, AutoPipelineForInpainting, AutoPipelineForImage2Image, DiffusionPipeline, StableDiffusionXLImg2ImgPipeline, StableDiffusionControlNetInpaintPipeline, ControlNetModel
4
  from PIL import Image
5
  import base64
6
  from io import BytesIO
@@ -32,7 +32,7 @@ class EndpointHandler():
32
  "runwayml/stable-diffusion-v1-5", controlnet=self.controlnet, torch_dtype=torch.float16
33
  )
34
 
35
- self.pipe.scheduler = EulerDiscreteScheduler.from_config(self.pipe.scheduler.config)
36
  self.pipe.enable_model_cpu_offload()
37
  self.pipe.enable_xformers_memory_efficient_attention()
38
 
 
1
  from typing import Dict, List, Any
2
  import torch
3
+ from diffusers import DPMSolverMultistepScheduler, EulerDiscreteScheduler, EulerAncestralDiscreteScheduler, DDIMScheduler, StableDiffusionInpaintPipeline, AutoPipelineForInpainting, AutoPipelineForImage2Image, DiffusionPipeline, StableDiffusionXLImg2ImgPipeline, StableDiffusionControlNetInpaintPipeline, ControlNetModel
4
  from PIL import Image
5
  import base64
6
  from io import BytesIO
 
32
  "runwayml/stable-diffusion-v1-5", controlnet=self.controlnet, torch_dtype=torch.float16
33
  )
34
 
35
+ self.pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(self.pipe.scheduler.config)
36
  self.pipe.enable_model_cpu_offload()
37
  self.pipe.enable_xformers_memory_efficient_attention()
38