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

Fix event args

Browse Source
This commit is contained in:
Sergey Borisov 2023-05-12 23:08:03 +03:00
parent 79fecba274
commit 5431dd5f50
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
invokeai/app/services/events.py
View File

@ -119,13 +119,13 @@ class EventServiceBase:
graph_execution_state_id=graph_execution_state_id, graph_execution_state_id=graph_execution_state_id,
node=node, node=node,
source_node_id=source_node_id, source_node_id=source_node_id,
model_name=str, model_name=model_name,
model_type=model_type, model_type=model_type.name,
submodel=submodel, submodel=submodel,
), ),
) )
def emit_model_load_completed ( def emit_model_load_completed(
self, self,
graph_execution_state_id: str, graph_execution_state_id: str,
node: dict, node: dict,
@ -137,13 +137,13 @@ class EventServiceBase:
) -> None: ) -> None:
"""Emitted when a model is correctly loaded (returns model info)""" """Emitted when a model is correctly loaded (returns model info)"""
self.__emit_session_event( self.__emit_session_event(
event_name="model_load_started", event_name="model_load_completed",
payload=dict( payload=dict(
graph_execution_state_id=graph_execution_state_id, graph_execution_state_id=graph_execution_state_id,
node=node, node=node,
source_node_id=source_node_id, source_node_id=source_node_id,
model_name=str, model_name=model_name,
model_type=model_type, model_type=model_type.name,
submodel=submodel, submodel=submodel,
model_info=model_info, model_info=model_info,
), ),