Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(app): iterate on processor split 2

Browse Source 
- Use protocol to define callbacks, this allows them to have kwargs
- Shuffle the profiler around a bit
- Move `thread_limit` and `polling_interval` to `__init__`; `start` is called programmatically and will never get these args in practice
This commit is contained in:
psychedelicious
2024-05-22 18:52:46 +10:00
parent be41c84305
commit 47722528a3
3 changed files with 80 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
invokeai/app/api/dependencies.py
View File

@ -112,7 +112,7 @@ class ApiDependencies:
print("BEFORE RUN NODE", invocation.id)
return True
def on_after_run_node(invocation, queue_item, outputs):
def on_after_run_node(invocation, queue_item, output):
print("AFTER RUN NODE", invocation.id)
return True
@ -124,17 +124,17 @@ class ApiDependencies:
print("AFTER RUN SESSION", queue_item.item_id)
return True
def on_non_fatal_processor_error(queue_item, exc_type, exc_value, exc_traceback):
def on_non_fatal_processor_error(exc_type, exc_value, exc_traceback, queue_item=None):
print("NON FATAL PROCESSOR ERROR", exc_value)
return True
session_processor = DefaultSessionProcessor(
DefaultSessionRunner(
on_before_run_session,
on_before_run_node,
on_after_run_node,
on_node_error,
on_after_run_session,
on_before_run_session=on_before_run_session,
on_before_run_node=on_before_run_node,
on_after_run_node=on_after_run_node,
on_node_error=on_node_error,
on_after_run_session=on_after_run_session,
),
on_non_fatal_processor_error,
)