Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

use AsyncTask for last_stop handling instead of shared

Browse Source
This commit is contained in:
Manuel Schmid
2023-11-18 17:20:41 +01:00
parent d3f9156854
commit 617255d040
3 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
webui.py
+4 -3
View File
@@ -20,12 +20,13 @@ from modules.private_logger import get_current_html_path
from modules.ui_gradio_extensions import reload_javascript
from modules.auth import auth_enabled, check_auth
currentTask = gr.State()
def generate_clicked(*args):
# outputs=[progress_html, progress_window, progress_gallery, gallery]
execution_start_time = time.perf_counter()
task = worker.AsyncTask(args=list(args))
currentTask.value = task = worker.AsyncTask(args=list(args))
finished = False
yield gr.update(visible=True, value=modules.html.make_progress_html(1, 'Waiting for task to start ...')), \
@@ -117,13 +118,13 @@ with shared.gradio_root:
def stop_clicked():
import fcbh.model_management as model_management
shared.last_stop = 'stop'
currentTask.value.last_stop = 'stop'
model_management.interrupt_current_processing()
return [gr.update(interactive=False)] * 2
def skip_clicked():
import fcbh.model_management as model_management
shared.last_stop = 'skip'
currentTask.value.last_stop = 'skip'
model_management.interrupt_current_processing()
return