Update config.json

config.json

@@ -1,19 +1,15 @@
-from diffusers import StableDiffusionPipeline
-import torch
-
-def init():
-    global model
-    model_id = "colt12/maxcushion"
-    model = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-    model.to("cuda")
-
-def inference(model_inputs):
-    global model
-    prompt = model_inputs.get('prompt', '')
-    image = model(prompt).images[0]
-    # Convert image to bytes
-    import io
-    buf = io.BytesIO()
-    image.save(buf, format='PNG')
-    byte_im = buf.getvalue()
-    return {"image_bytes": byte_im}
+{
+  "architectures": ["UNet2DConditionModel"],
+  "_class_name": "StableDiffusionPipeline",
+  "_diffusers_version": "0.21.4",
+  "feature_extractor": [
+    "CLIPFeatureExtractor",
+    "openai/clip-vit-base-patch32"
+  ],
+  "safety_checker": null,
+  "scheduler": ["PNDMScheduler"],
+  "text_encoder": ["CLIPTextModel", "openai/clip-vit-base-patch32"],
+  "tokenizer": ["CLIPTokenizer", "openai/clip-vit-base-patch32"],
+  "unet": ["UNet2DConditionModel"],
+  "vae": ["AutoencoderKL"]
+}