what

handler.py

@@ -15,6 +15,7 @@ model_id = "stabilityai/stable-diffusion-2-1-base"
 
 class EndpointHandler():
     def __init__(self, path=""):
+        raise NotImplementedError();
         # load the optimized model
         self.textPipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
         self.textPipe.scheduler = DDIMScheduler.from_config(self.textPipe.scheduler.config)
@@ -36,7 +37,6 @@ class EndpointHandler():
         image_data = data.pop("inputs", data).decode('base64')
         init_image = Image.open(image_data).convert("RGB")
         init_image.thumbnail((512, 512))
-        raise NotImplementedError()
 
         params = data.pop("parameters", data)