Update __init__.py

shadowcz007 · Sep 1, 2024 · 078fd10 · 078fd10
1 parent 824e25d
commit 078fd10
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/__init__.py b/__init__.py
@@ -983,7 +983,9 @@ async def handle_ar_page(request):
 # 重启服务
 @routes.post('/mixlab/re_start')
 def re_start(request):
+    p_intance=PromptServer.instance
     try:
+        p_intance.prompt_queue.set_flag("free_memory", True)
         sys.stdout.close_log()
     except Exception as e:
         pass