Update handler.py
Browse files- handler.py +5 -5
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 |
-
print("Logs: self.multi_text2image_model", multi_text2image_model)
|
| 204 |
prompt_modifier.load()
|
| 205 |
|
| 206 |
lora_style.load(path)
|
|
@@ -225,10 +225,10 @@ 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["model_id"])
|
| 229 |
-
print("Logs: multimodel controlnet pipelines are",model, multi_controlnet_model
|
| 230 |
-
print("Logs: multimodel text2img pipelines are",model, multi_text2image_model
|
| 231 |
-
print("Logs: multimodel imgtoimage pipelines are",model, multi_image2image_model
|
| 232 |
# controlnet.load(path)
|
| 233 |
# text2img_pipe.load(path)
|
| 234 |
# img2img_pipe.load(path)
|
|
|
|
| 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)
|
|
|
|
| 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)
|
| 230 |
+
print("Logs: multimodel text2img pipelines are",model, multi_text2image_model)
|
| 231 |
+
print("Logs: multimodel imgtoimage pipelines are",model, multi_image2image_model)
|
| 232 |
# controlnet.load(path)
|
| 233 |
# text2img_pipe.load(path)
|
| 234 |
# img2img_pipe.load(path)
|