Upload 4 files

config.json

@@ -0,0 +1,5 @@
+{
+  "_class_name": "UNet2DConditionModel",
+  "_diffusers_version": "0.16.1",
+  "pipeline_tag": "stable-diffusion-image-to-image"
+}

handler.py

@@ -0,0 +1,35 @@
+import base64
+import io
+from PIL import Image
+import torch
+from diffusers import StableDiffusionImg2ImgPipeline
+
+pipe = None
+
+class EndpointHandler:
+    def __init__(self):
+        self.device = "cuda" if torch.cuda.is_available() else "cpu"
+
+    def init(self):
+        global pipe
+        pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
+            "InstantX/InstantID",
+            torch_dtype=torch.float16,
+            safety_checker=None
+        ).to(self.device)
+        pipe.enable_attention_slicing()
+
+    def inference(self, model_inputs: dict) -> dict:
+        img_bytes = base64.b64decode(model_inputs.get("image_base64"))
+        init_image = Image.open(io.BytesIO(img_bytes)).convert("RGB")
+        output = pipe(
+            prompt=model_inputs.get("prompt", ""),
+            image=init_image,
+            strength=float(model_inputs.get("strength", 0.75)),
+            guidance_scale=float(model_inputs.get("guidance_scale", 7.5)),
+            num_inference_steps=int(model_inputs.get("num_inference_steps", 50)),
+        )
+        sticker = output.images[0]
+        buf = io.BytesIO()
+        sticker.save(buf, format="PNG")
+        return {"generated_image_base64": base64.b64encode(buf.getvalue()).decode("utf-8")}

model_index.json

@@ -0,0 +1,22 @@
+{
+  "_class_name": "StableDiffusionImg2ImgPipeline",
+  "_diffusers_version": "0.16.1",
+  "components": {
+    "unet": {
+      "path": "diffusers/unet",
+      "type": ["diffusers", "UNet2DConditionModel"]
+    },
+    "vae": {
+      "path": "diffusers/vae",
+      "type": ["diffusers", "AutoencoderKL"]
+    },
+    "scheduler": {
+      "path": "diffusers/scheduler",
+      "type": ["diffusers", "DDIMScheduler"]
+    },
+    "tokenizer": {
+      "path": "tokenizer",
+      "type": ["transformers", "CLIPTokenizer"]
+    }
+  }
+}

requirements.txt

@@ -0,0 +1,6 @@
+torch>=2.0.0
+diffusers>=0.16.1
+transformers>=4.30.0
+huggingface_hub>=0.14.1
+safetensors
+Pillow