Update handler.py
Browse files- handler.py +1 -0
handler.py
CHANGED
|
@@ -10,6 +10,7 @@ from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
|
|
| 10 |
|
| 11 |
class EndpointHandler():
|
| 12 |
def __init__(self, path="", model_base="runwayml/stable-diffusion-v1-5"):
|
|
|
|
| 13 |
self.pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16)
|
| 14 |
self.pipe.scheduler = DPMSolverMultistepScheduler.from_config(self.pipe.scheduler.config)
|
| 15 |
self.original_unet = copy.deepcopy(self.pipe.unet)
|
|
|
|
| 10 |
|
| 11 |
class EndpointHandler():
|
| 12 |
def __init__(self, path="", model_base="runwayml/stable-diffusion-v1-5"):
|
| 13 |
+
print("diffusers.__version__: " + diffusers.__version__)
|
| 14 |
self.pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16)
|
| 15 |
self.pipe.scheduler = DPMSolverMultistepScheduler.from_config(self.pipe.scheduler.config)
|
| 16 |
self.original_unet = copy.deepcopy(self.pipe.unet)
|