Update handler.py
Browse files- handler.py +2 -4
handler.py
CHANGED
|
@@ -189,7 +189,7 @@ if device.type != 'cuda':
|
|
| 189 |
multi_model_list = [
|
| 190 |
{"model_id": "jayparmr/icbinp"},
|
| 191 |
{"model_id": "jayparmr/comic_anime"},
|
| 192 |
-
{"model_id": "
|
| 193 |
]
|
| 194 |
|
| 195 |
multi_controlnet_model={}
|
|
@@ -211,9 +211,7 @@ class EndpointHandler():
|
|
| 211 |
for model in multi_model_list:
|
| 212 |
print("Logs: model value is", model)
|
| 213 |
print("Logs: model path value is", model["model_id"] )
|
| 214 |
-
|
| 215 |
-
# self.multi_text2image_model[model["model_id"]] = text2img_pipe.load(model["model_id"])
|
| 216 |
-
# self.multi_image2image_model[model["model_id"]] = img2img_pipe.load(model["model_id"])
|
| 217 |
controlnet = ControlNet()
|
| 218 |
img2img_pipe = Img2Img()
|
| 219 |
text2img_pipe = Text2Img()
|
|
|
|
| 189 |
multi_model_list = [
|
| 190 |
{"model_id": "jayparmr/icbinp"},
|
| 191 |
{"model_id": "jayparmr/comic_anime"},
|
| 192 |
+
{"model_id": "jayparmr/icbinp"},
|
| 193 |
]
|
| 194 |
|
| 195 |
multi_controlnet_model={}
|
|
|
|
| 211 |
for model in multi_model_list:
|
| 212 |
print("Logs: model value is", model)
|
| 213 |
print("Logs: model path value is", model["model_id"] )
|
| 214 |
+
|
|
|
|
|
|
|
| 215 |
controlnet = ControlNet()
|
| 216 |
img2img_pipe = Img2Img()
|
| 217 |
text2img_pipe = Text2Img()
|