diff --git a/BaiHu/Util/__pycache__/SDUtil.cpython-310.pyc b/BaiHu/Util/__pycache__/SDUtil.cpython-310.pyc
index e64cbefe..bdb5c391 100644
Binary files a/BaiHu/Util/__pycache__/SDUtil.cpython-310.pyc and b/BaiHu/Util/__pycache__/SDUtil.cpython-310.pyc differ
diff --git a/BaiHu/doTask.py b/BaiHu/doTask.py
index d55504ee..bacec740 100644
--- 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()
             # 回复上位生成完毕