Update handler.py
Browse files- 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
|
| 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 |
|