jayparmr commited on
Commit
2068416
·
1 Parent(s): 1625be6

Update handler.py

Browse files
Files changed (1) hide show
  1. 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
- print("Logs: model value is", model)
213
- print("Logs: model path value is", model["model_id"] )
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",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