diff --git a/BaiHu/doTask.py b/BaiHu/doTask.py
index 2b7ca677..827ffc1f 100644
--- a/BaiHu/doTask.py
+++ b/BaiHu/doTask.py
@@ -134,7 +134,6 @@ def download_image(image_url, file_path):
         printf('图片下载失败，状态码：' + str(response.status_code))
 
 
-
 # 指定目录下有多少个指定前缀的文件
 def getPrefixFileCount(directory_path, prefix):
     # 使用os.listdir()获取目录下的所有文件和文件夹
@@ -278,13 +277,13 @@ def get_task():
             if model_type_id == 1:
                 webui_txt_2_img(model_id=model_id, json_data=vjson_data, input_image=input_image)
 
-            # COMFY_UI
-            if model_type_id == 2:
-                runComfyUI(model_id=model_id, json_data=vjson_data, input_image=input_image)
-
             # SD_IMG2IMG
+            if model_type_id == 2:
+                webui_img_2_img(model_id=model_id, prompt_id=prompt_id, json_data=vjson_data,
+                                input_image=input_image)
+            # COMFY_UI
             if model_type_id == 3:
-                webui_img_2_img(model_id=model_id, prompt_id=prompt_id, json_data=vjson_data, input_image=input_image)
+                runComfyUI(model_id=model_id, json_data=vjson_data, input_image=input_image)
 
             return task_id
         elif success == 0: