karimbenharrak
commited on
Commit
•
43a4c7b
1
Parent(s):
13715fa
Update handler.py
Browse files- 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, 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 =
|
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, 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 = EulerDiscreteScheduler.from_config(self.pipe.scheduler.config)
|
36 |
self.pipe.enable_model_cpu_offload()
|
37 |
self.pipe.enable_xformers_memory_efficient_attention()
|
38 |
|