'commit'

1 year ago · 20ac7c4907
parent 58ee017fca
commit 20ac7c4907
2 changed files with 3 additions and 3 deletions
--- a/BaiHu/Util/pycache/SDUtil.cpython-310.pyc
+++ b/BaiHu/Util/pycache/SDUtil.cpython-310.pyc
--- a/BaiHu/doTask.py
+++ b/BaiHu/doTask.py
@ -198,6 +198,9 @@ def get_task():
    if response.status_code == 200:
        success = response.json()["success"]
        if success == 1:
+            # 清理服务器显存
+            release_sd(webui_address, comfyui_address)
+
            data = response.json()["data"]
            # 任务编号
            task_id = data['task_id']
@ -314,9 +317,6 @@ if __name__ == '__main__':

    while True:
        try:
-            # 清理服务器显存
-            release_sd(webui_address, comfyui_address)
-
            res = []  # 所有图片的地址路径
            v_task_id = get_task()
            # 回复上位生成完毕