jayparmr commited on
Commit
9b49fa9
·
1 Parent(s): 286a4d3

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +3 -3
handler.py CHANGED
@@ -128,7 +128,7 @@ def pose(task: Task, s3_outkey: str = "_pose", poses: Optional[list] = None):
128
  @slack.auto_send_alert
129
  def text2img(task: Task, text2img_pipe ):
130
  prompt, ori_prompt = get_patched_prompt(task)
131
-
132
  lora_patcher = lora_style.get_patcher(text2img_pipe.pipe, task.get_style())
133
  lora_patcher.patch()
134
 
@@ -270,7 +270,7 @@ class EndpointHandler():
270
 
271
  try:
272
  task_type = task.get_type()
273
-
274
  if task_type == TaskType.TEXT_TO_IMAGE:
275
  # character sheet
276
  if "character sheet" in task.get_prompt().lower():
@@ -288,7 +288,7 @@ class EndpointHandler():
288
  except Exception as e:
289
  print(f"Error: {e}")
290
  slack.error_alert(data, e)
291
- return { "error": e, "data":data, "task_type":task_type, "self.multi_text2image_model[model_id]":self.multi_text2image_model[model_id] }
292
 
293
  # inputs = data.pop("inputs", data)
294
 
 
128
  @slack.auto_send_alert
129
  def text2img(task: Task, text2img_pipe ):
130
  prompt, ori_prompt = get_patched_prompt(task)
131
+ print("logs post: text2img_pipe", text2img_pipe)
132
  lora_patcher = lora_style.get_patcher(text2img_pipe.pipe, task.get_style())
133
  lora_patcher.patch()
134
 
 
270
 
271
  try:
272
  task_type = task.get_type()
273
+ print("logs post: self.multi_text2image_model[model_id]", self.multi_text2image_model[model_id])
274
  if task_type == TaskType.TEXT_TO_IMAGE:
275
  # character sheet
276
  if "character sheet" in task.get_prompt().lower():
 
288
  except Exception as e:
289
  print(f"Error: {e}")
290
  slack.error_alert(data, e)
291
+ return { "error": e, "data":data, "task_type":task_type }
292
 
293
  # inputs = data.pop("inputs", data)
294