diff --git a/BaiHu/doTask.py b/BaiHu/doTask.py
index bacec740..3367103e 100644
--- a/BaiHu/doTask.py
+++ b/BaiHu/doTask.py
@@ -258,6 +258,9 @@ def get_task():
                 runComfyUI(model_id=model_id, json_data=vjson_data, input_image=input_image,
                            target_folder=output_path)
 
+            # 清理服务器显存
+            release_sd(webui_address, comfyui_address)
+
             return task_id
         elif success == 0:
             printf("现在没有待处理的任务,休息3秒后再试...")