add restoration services to nodes

This commit is contained in:
Lincoln Stein
2023-03-11 17:00:00 -05:00
parent 3aa1ee1218
commit 8ca91b1774
9 changed files with 125 additions and 8 deletions

View File

@ -5,6 +5,7 @@ from argparse import Namespace
from ...backend import Globals
from ..services.model_manager_initializer import get_model_manager
from ..services.restoration_services import RestorationServices
from ..services.graph import GraphExecutionState
from ..services.image_storage import DiskImageStorage
from ..services.invocation_queue import MemoryInvocationQueue
@ -67,6 +68,7 @@ class ApiDependencies:
filename=db_location, table_name="graph_executions"
),
processor=DefaultInvocationProcessor(),
restoration=RestorationServices(config),
)
ApiDependencies.invoker = Invoker(services)