Update handler.py
Browse files- handler.py +6 -6
handler.py
CHANGED
|
@@ -188,9 +188,9 @@ if device.type != 'cuda':
|
|
| 188 |
|
| 189 |
# multi-model list
|
| 190 |
multi_model_list = [
|
| 191 |
-
{"model_id": "
|
| 192 |
-
|
| 193 |
-
|
| 194 |
]
|
| 195 |
|
| 196 |
class EndpointHandler():
|
|
@@ -213,9 +213,9 @@ class EndpointHandler():
|
|
| 213 |
# self.multi_controlnet_model[model["model_id"]] = controlnet.load(model["model_id"])
|
| 214 |
# self.multi_text2image_model[model["model_id"]] = text2img_pipe.load(model["model_id"])
|
| 215 |
# self.multi_image2image_model[model["model_id"]] = img2img_pipe.load(model["model_id"])
|
| 216 |
-
self.multi_controlnet_model[model["model_id"]] = controlnet.load(
|
| 217 |
-
self.multi_text2image_model[model["model_id"]] = text2img_pipe.load(
|
| 218 |
-
self.multi_image2image_model[model["model_id"]] = img2img_pipe.load(
|
| 219 |
|
| 220 |
print(" Logs: model[model_id]", model["model_id"])
|
| 221 |
print("Logs: multimodel controlnet pipelines are", path + model["model_id"])
|
|
|
|
| 188 |
|
| 189 |
# multi-model list
|
| 190 |
multi_model_list = [
|
| 191 |
+
{"model_id": "jayparmr/icbinp"},
|
| 192 |
+
{"model_id": "Bingsu/my-korean-stable-diffusion-v1-5"},
|
| 193 |
+
{"model_id": "jcplus/stable-diffusion-v1-5"},
|
| 194 |
]
|
| 195 |
|
| 196 |
class EndpointHandler():
|
|
|
|
| 213 |
# self.multi_controlnet_model[model["model_id"]] = controlnet.load(model["model_id"])
|
| 214 |
# self.multi_text2image_model[model["model_id"]] = text2img_pipe.load(model["model_id"])
|
| 215 |
# self.multi_image2image_model[model["model_id"]] = img2img_pipe.load(model["model_id"])
|
| 216 |
+
self.multi_controlnet_model[model["model_id"]] = controlnet.load(model["model_id"])
|
| 217 |
+
self.multi_text2image_model[model["model_id"]] = text2img_pipe.load( model["model_id"])
|
| 218 |
+
self.multi_image2image_model[model["model_id"]] = img2img_pipe.load( model["model_id"])
|
| 219 |
|
| 220 |
print(" Logs: model[model_id]", model["model_id"])
|
| 221 |
print("Logs: multimodel controlnet pipelines are", path + model["model_id"])
|