diff --git a/BaiHu/doTask.py b/BaiHu/doTask.py index eba0b619..ddf3be0a 100644 --- a/BaiHu/doTask.py +++ b/BaiHu/doTask.py @@ -247,7 +247,6 @@ def get_task(): source_img_url = data["source_img_url"] # 模板类型 model_type_id = data["model_type_id"] - # json文件 json_file = r'./JSON/' + str(model_id) + '/' + str(model_id) + '_' + str(prompt_id) + '.json' @@ -270,9 +269,10 @@ def get_task(): vjson_data = json.load(file) # 利用三元表达式输出 - printf("发现需要处理的" + ("WebUI" if model_type_id in [1, 2] else "ComfyUI") + "任务,model_type_id=" + str( - model_type_id) + ",model_id=" + str( - model_id) + ",prompt_id=" + str(prompt_id) + ",task_id=" + str(task_id)) + printf( + "发现需要处理的" + ("WebUI" if model_type_id in [1, 2, 4] else "ComfyUI") + "任务,model_type_id=" + str( + model_type_id) + ",model_id=" + str( + model_id) + ",prompt_id=" + str(prompt_id) + ",task_id=" + str(task_id)) # 创建目标目录 output_path = "Out/Images/User/" + str(model_id) + "/"