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

docs: tidy comments in processor

Browse Source
This commit is contained in:
psychedelicious 2024-04-01 16:59:26 +11:00
parent 7b93cc8538
commit 9aeabf10df
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
invokeai/app/services/session_processor/session_processor_default.py
View File

@ -157,7 +157,6 @@ class DefaultSessionProcessor(SessionProcessorBase):
while self._invocation is not None and not cancel_event.is_set(): while self._invocation is not None and not cancel_event.is_set():
# get the source node id to provide to clients (the prepared node id is not as useful) # get the source node id to provide to clients (the prepared node id is not as useful)
source_invocation_id = self._queue_item.session.prepared_source_mapping[self._invocation.id] source_invocation_id = self._queue_item.session.prepared_source_mapping[self._invocation.id]
self._invoker.services.events.emit_invocation_started(self._queue_item, self._invocation) self._invoker.services.events.emit_invocation_started(self._queue_item, self._invocation)
# Innermost processor try block; any unhandled exception is an invocation error & will fail the graph # Innermost processor try block; any unhandled exception is an invocation error & will fail the graph
@ -185,13 +184,12 @@ class DefaultSessionProcessor(SessionProcessorBase):
# Save outputs and history # Save outputs and history
self._queue_item.session.complete(self._invocation.id, outputs) self._queue_item.session.complete(self._invocation.id, outputs)
# Dispatch invocation complete event
self._invoker.services.events.emit_invocation_complete( self._invoker.services.events.emit_invocation_complete(
self._queue_item, self._invocation, outputs self._queue_item, self._invocation, outputs
) )
except KeyboardInterrupt: except KeyboardInterrupt:
# TODO(MM2): Create an event for this # TODO(MM2): I don't think this is ever raised...
pass pass
except CanceledException: except CanceledException:
@ -217,7 +215,6 @@ class DefaultSessionProcessor(SessionProcessorBase):
) )
self._invoker.services.logger.error(error) self._invoker.services.logger.error(error)
# Send error event
self._invoker.services.events.emit_invocation_error( self._invoker.services.events.emit_invocation_error(
queue_item=self._queue_item, queue_item=self._queue_item,
invocation=self._invocation, invocation=self._invocation,
@ -230,7 +227,6 @@ class DefaultSessionProcessor(SessionProcessorBase):
# The session is complete if the all invocations are complete or there was an error # The session is complete if the all invocations are complete or there was an error
if self._queue_item.session.is_complete() or cancel_event.is_set(): if self._queue_item.session.is_complete() or cancel_event.is_set():
# Send complete event
self._invoker.services.events.emit_session_complete(self._queue_item) self._invoker.services.events.emit_session_complete(self._queue_item)
# If we are profiling, stop the profiler and dump the profile & stats # If we are profiling, stop the profiler and dump the profile & stats
if self._profiler: if self._profiler: