|
|
|
|
@ -215,7 +215,8 @@ def runComfyUI(model_id, json_data, input_image, target_folder):
|
|
|
|
|
# # 生成下载链接
|
|
|
|
|
url = 'http://hzkc.oss-cn-beijing.aliyuncs.com/' + key
|
|
|
|
|
res.append(url)
|
|
|
|
|
|
|
|
|
|
# 清理GPU缓存
|
|
|
|
|
clear_cache(server_address)
|
|
|
|
|
|
|
|
|
|
# 获取任务
|
|
|
|
|
def get_task():
|
|
|
|
|
@ -289,8 +290,7 @@ def get_task():
|
|
|
|
|
if model_type_id == 3:
|
|
|
|
|
runComfyUI(model_id=model_id, json_data=vjson_data, input_image=input_image,
|
|
|
|
|
target_folder=output_path)
|
|
|
|
|
# 清理GPU缓存
|
|
|
|
|
clear_cache(server_address)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
return task_id
|
|
|
|
|
elif success == 0:
|
|
|
|
|
|
