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

Instantiate batch managers

Browse Source
This commit is contained in:
Brandon Rising 2023-08-01 16:44:17 -04:00
parent a61685696f
commit a3e099bbc0
3 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/app/cli_app.py
View File

@ -37,6 +37,7 @@ from invokeai.app.services.image_record_storage import SqliteImageRecordStorage
from invokeai.app.services.images import ImageService, ImageServiceDependencies from invokeai.app.services.images import ImageService, ImageServiceDependencies
from invokeai.app.services.resource_name import SimpleNameService from invokeai.app.services.resource_name import SimpleNameService
from invokeai.app.services.urls import LocalUrlService from invokeai.app.services.urls import LocalUrlService
from invokeai.app.services.batch_manager import BatchManager
from .services.default_graphs import default_text_to_image_graph_id, create_system_graphs from .services.default_graphs import default_text_to_image_graph_id, create_system_graphs
from .services.latent_storage import DiskLatentsStorage, ForwardCacheLatentsStorage from .services.latent_storage import DiskLatentsStorage, ForwardCacheLatentsStorage
@ -306,6 +307,7 @@ def invoke_cli():
latents=ForwardCacheLatentsStorage(DiskLatentsStorage(f"{output_folder}/latents")), latents=ForwardCacheLatentsStorage(DiskLatentsStorage(f"{output_folder}/latents")),
images=images, images=images,
boards=boards, boards=boards,
batch_manager=BatchManager(),
board_images=board_images, board_images=board_images,
queue=MemoryInvocationQueue(), queue=MemoryInvocationQueue(),
graph_library=SqliteItemStorage[LibraryGraph](filename=db_location, table_name="graphs"), graph_library=SqliteItemStorage[LibraryGraph](filename=db_location, table_name="graphs"),

1
tests/nodes/test_graph_execution_state.py
View File

@ -48,6 +48,7 @@ def mock_services() -> InvocationServices:
images=None, # type: ignore images=None, # type: ignore
latents=None, # type: ignore latents=None, # type: ignore
boards=None, # type: ignore boards=None, # type: ignore
batch_manager=None, # type: ignore
board_images=None, # type: ignore board_images=None, # type: ignore
queue=MemoryInvocationQueue(), queue=MemoryInvocationQueue(),
graph_library=SqliteItemStorage[LibraryGraph](filename=sqlite_memory, table_name="graphs"), graph_library=SqliteItemStorage[LibraryGraph](filename=sqlite_memory, table_name="graphs"),

1
tests/nodes/test_invoker.py
View File

@ -40,6 +40,7 @@ def mock_services() -> InvocationServices:
logger=None, # type: ignore logger=None, # type: ignore
images=None, # type: ignore images=None, # type: ignore
latents=None, # type: ignore latents=None, # type: ignore
batch_manager=None, # type: ignore
boards=None, # type: ignore boards=None, # type: ignore
board_images=None, # type: ignore board_images=None, # type: ignore
queue=MemoryInvocationQueue(), queue=MemoryInvocationQueue(),