'commit'

1 year ago · f7d65ff683
parent c6d04ad688
commit f7d65ff683
2 changed files with 3 additions and 2 deletions
--- a/BaiHu/Util/ComfyUIUtil.py
+++ b/BaiHu/Util/ComfyUIUtil.py
@ -138,6 +138,8 @@ def generate_clip(server_address, prompt_data, client_id, output_path, myfilter)
                # 写入二进制文件
                binary_file.write(image_data)
    ws.close()
+    # 清理GPU缓存
+    clear_cache(server_address)
    return files


--- a/BaiHu/doTask.py
+++ b/BaiHu/doTask.py
@ -215,8 +215,7 @@ 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():