Update handler.py
Browse files- handler.py +10 -10
handler.py
CHANGED
|
@@ -200,7 +200,7 @@ class EndpointHandler():
|
|
| 200 |
def __init__(self, path=""):
|
| 201 |
# load the optimized model
|
| 202 |
print("Logs: model loaded .... starts")
|
| 203 |
-
|
| 204 |
prompt_modifier.load()
|
| 205 |
|
| 206 |
lora_style.load(path)
|
|
@@ -209,8 +209,8 @@ class EndpointHandler():
|
|
| 209 |
self.path = path
|
| 210 |
|
| 211 |
for model in multi_model_list:
|
| 212 |
-
|
| 213 |
-
|
| 214 |
|
| 215 |
controlnet = ControlNet()
|
| 216 |
img2img_pipe = Img2Img()
|
|
@@ -225,16 +225,16 @@ class EndpointHandler():
|
|
| 225 |
multi_image2image_model[model["model_id"]] = img2img_pipe;
|
| 226 |
img2img_pipe.load( model["model_id"])
|
| 227 |
|
| 228 |
-
print(" Logs: model[model_id]", model["model_id"])
|
| 229 |
-
print("Logs: multimodel controlnet pipelines are", multi_controlnet_model[model["model_id"]])
|
| 230 |
-
print("Logs: multimodel text2img pipelines are", multi_text2image_model[model["model_id"]])
|
| 231 |
-
print("Logs: multimodel imgtoimage pipelines are", multi_image2image_model[model["model_id"]])
|
| 232 |
# controlnet.load(path)
|
| 233 |
# text2img_pipe.load(path)
|
| 234 |
# img2img_pipe.load(path)
|
| 235 |
-
print("Logs: self.multi_image2image_model"
|
| 236 |
-
print("Logs: self.multi_text2image_model",multi_text2image_model)
|
| 237 |
-
print("Logs: self.multi_controlnet_model",multi_controlnet_model)
|
| 238 |
print("Logs: model loaded ....")
|
| 239 |
|
| 240 |
|
|
|
|
| 200 |
def __init__(self, path=""):
|
| 201 |
# load the optimized model
|
| 202 |
print("Logs: model loaded .... starts")
|
| 203 |
+
print("Logs: self.multi_text2image_model", multi_text2image_model)
|
| 204 |
prompt_modifier.load()
|
| 205 |
|
| 206 |
lora_style.load(path)
|
|
|
|
| 209 |
self.path = path
|
| 210 |
|
| 211 |
for model in multi_model_list:
|
| 212 |
+
|
| 213 |
+
|
| 214 |
|
| 215 |
controlnet = ControlNet()
|
| 216 |
img2img_pipe = Img2Img()
|
|
|
|
| 225 |
multi_image2image_model[model["model_id"]] = img2img_pipe;
|
| 226 |
img2img_pipe.load( model["model_id"])
|
| 227 |
|
| 228 |
+
print(" Logs: model[model_id]",model, model["model_id"])
|
| 229 |
+
print("Logs: multimodel controlnet pipelines are",model, multi_controlnet_model[model["model_id"]])
|
| 230 |
+
print("Logs: multimodel text2img pipelines are",model, multi_text2image_model[model["model_id"]])
|
| 231 |
+
print("Logs: multimodel imgtoimage pipelines are",model, multi_image2image_model[model["model_id"]])
|
| 232 |
# controlnet.load(path)
|
| 233 |
# text2img_pipe.load(path)
|
| 234 |
# img2img_pipe.load(path)
|
| 235 |
+
print("Logs: self.multi_image2image_model")
|
| 236 |
+
print("Logs: self.multi_text2image_model", multi_text2image_model)
|
| 237 |
+
print("Logs: self.multi_controlnet_model", multi_controlnet_model)
|
| 238 |
print("Logs: model loaded ....")
|
| 239 |
|
| 240 |
|