mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
21 lines
612 B
Python
21 lines
612 B
Python
|
# Copyright (c) 2022 Kyle Schouviller (https://github.com/kyle0654)
|
||
|
|
from .image_storage import ImageStorageBase
|
||
|
|
from .events import EventServiceBase
|
||
|
|
from ....generate import Generate
|
||
|
|
|
||
|
|
|
||
|
|
class InvocationServices():
|
||
|
|
"""Services that can be used by invocations"""
|
||
|
|
generate: Generate # TODO: wrap Generate, or split it up from model?
|
||
|
|
events: EventServiceBase
|
||
|
|
images: ImageStorageBase
|
||
|
|
|
||
|
|
def __init__(self,
|
||
|
|
generate: Generate,
|
||
|
|
events: EventServiceBase,
|
||
|
|
images: ImageStorageBase
|
||
|
|
):
|
||
|
|
self.generate = generate
|
||
|
|
self.events = events
|
||
|
|
self.images = images
|