Update handler.py

handler.py

@@ -126,7 +126,7 @@ def pose(task: Task, s3_outkey: str = "_pose", poses: Optional[list] = None):
 # @update_db
 @auto_clear_cuda_and_gc(controlnet)
 @slack.auto_send_alert
-def text2img(task: Task):
+def text2img(task: Task, text2img_pipe ):
     prompt, ori_prompt = get_patched_prompt(task)
 
     lora_patcher = lora_style.get_patcher(text2img_pipe.pipe, task.get_style())
@@ -243,7 +243,7 @@ class EndpointHandler():
         model_id = data.pop("model_id", None)
 
         # check if model_id is in the list of models
-        if model_id is None or model_id not in self.multi_model:
+        if model_id is None or model_id not in self.multi_model_list:
             raise ValueError(f"model_id: {model_id} is not valid. Available models are: {list(self.multi_model.keys())}")
 
         # # pass inputs with all kwargs in data
@@ -263,7 +263,7 @@ class EndpointHandler():
         parameters = data.pop("parameters", None)
         model_id = data.pop("model_id", None)
         
-        model_id ="" 
+        
         print("Logs post: model_id is", model_id)
         task = Task(data)
         
@@ -276,7 +276,7 @@ class EndpointHandler():
                 if "character sheet" in task.get_prompt().lower():
                     return pose(task, s3_outkey="", poses=pickPoses())
                 else:
-                    return self.multi_text2image_model[model_id](task)
+                    return text2img(task, self.multi_text2image_model[model_id])
             elif task_type == TaskType.IMAGE_TO_IMAGE:
                 return img2img(task)
             elif task_type == TaskType.CANNY: