'commit'

1 year ago · 6bbf81f3f1
parent 5fd50d3b89
commit 6bbf81f3f1
1 changed files with 2 additions and 0 deletions
--- a/BaiHu/doTask.py
+++ b/BaiHu/doTask.py
@ -289,6 +289,8 @@ 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: