mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
130 lines
3.7 KiB
Python
130 lines
3.7 KiB
Python
"""
|
|
This file re-exports all the public API for invocations. This is the only file that should be imported by custom nodes.
|
|
|
|
TODO(psyche): Do we want to dogfood this?
|
|
"""
|
|
|
|
from invokeai.app.invocations.baseinvocation import (
|
|
BaseInvocation,
|
|
BaseInvocationOutput,
|
|
invocation,
|
|
invocation_output,
|
|
)
|
|
from invokeai.app.invocations.fields import (
|
|
BoardField,
|
|
ColorField,
|
|
ConditioningField,
|
|
DenoiseMaskField,
|
|
FieldDescriptions,
|
|
FieldKind,
|
|
ImageField,
|
|
Input,
|
|
InputField,
|
|
LatentsField,
|
|
MetadataField,
|
|
OutputField,
|
|
UIComponent,
|
|
UIType,
|
|
WithMetadata,
|
|
WithWorkflow,
|
|
)
|
|
from invokeai.app.invocations.primitives import (
|
|
BooleanCollectionOutput,
|
|
BooleanOutput,
|
|
ColorCollectionOutput,
|
|
ColorOutput,
|
|
ConditioningCollectionOutput,
|
|
ConditioningOutput,
|
|
DenoiseMaskOutput,
|
|
FloatCollectionOutput,
|
|
FloatOutput,
|
|
ImageCollectionOutput,
|
|
ImageOutput,
|
|
IntegerCollectionOutput,
|
|
IntegerOutput,
|
|
LatentsCollectionOutput,
|
|
LatentsOutput,
|
|
StringCollectionOutput,
|
|
StringOutput,
|
|
)
|
|
from invokeai.app.services.boards.boards_common import BoardDTO
|
|
from invokeai.app.services.config.config_default import InvokeAIAppConfig
|
|
from invokeai.app.services.image_records.image_records_common import ImageCategory
|
|
from invokeai.app.services.shared.invocation_context import InvocationContext
|
|
from invokeai.app.services.workflow_records.workflow_records_common import WorkflowWithoutID
|
|
from invokeai.backend.model_management.model_manager import ModelInfo
|
|
from invokeai.backend.model_management.models.base import BaseModelType, ModelType, SubModelType
|
|
from invokeai.backend.stable_diffusion.diffusers_pipeline import PipelineIntermediateState
|
|
from invokeai.backend.stable_diffusion.diffusion.conditioning_data import (
|
|
BasicConditioningInfo,
|
|
ConditioningFieldData,
|
|
ExtraConditioningInfo,
|
|
SDXLConditioningInfo,
|
|
)
|
|
|
|
__all__ = [
|
|
# invokeai.app.invocations.baseinvocation
|
|
"BaseInvocation",
|
|
"BaseInvocationOutput",
|
|
"invocation",
|
|
"invocation_output",
|
|
# invokeai.app.services.shared.invocation_context
|
|
"InvocationContext",
|
|
# invokeai.app.invocations.fields
|
|
"BoardField",
|
|
"ColorField",
|
|
"ConditioningField",
|
|
"DenoiseMaskField",
|
|
"FieldDescriptions",
|
|
"FieldKind",
|
|
"ImageField",
|
|
"Input",
|
|
"InputField",
|
|
"LatentsField",
|
|
"MetadataField",
|
|
"OutputField",
|
|
"UIComponent",
|
|
"UIType",
|
|
"WithMetadata",
|
|
"WithWorkflow",
|
|
# invokeai.app.invocations.primitives
|
|
"BooleanCollectionOutput",
|
|
"BooleanOutput",
|
|
"ColorCollectionOutput",
|
|
"ColorOutput",
|
|
"ConditioningCollectionOutput",
|
|
"ConditioningOutput",
|
|
"DenoiseMaskOutput",
|
|
"FloatCollectionOutput",
|
|
"FloatOutput",
|
|
"ImageCollectionOutput",
|
|
"ImageOutput",
|
|
"IntegerCollectionOutput",
|
|
"IntegerOutput",
|
|
"LatentsCollectionOutput",
|
|
"LatentsOutput",
|
|
"StringCollectionOutput",
|
|
"StringOutput",
|
|
# invokeai.app.services.image_records.image_records_common
|
|
"ImageCategory",
|
|
# invokeai.app.services.boards.boards_common
|
|
"BoardDTO",
|
|
# invokeai.backend.stable_diffusion.diffusion.conditioning_data
|
|
"BasicConditioningInfo",
|
|
"ConditioningFieldData",
|
|
"ExtraConditioningInfo",
|
|
"SDXLConditioningInfo",
|
|
# invokeai.backend.stable_diffusion.diffusers_pipeline
|
|
"PipelineIntermediateState",
|
|
# invokeai.app.services.workflow_records.workflow_records_common
|
|
"WorkflowWithoutID",
|
|
# invokeai.app.services.config.config_default
|
|
"InvokeAIAppConfig",
|
|
# invokeai.backend.model_management.model_manager
|
|
"ModelInfo",
|
|
# invokeai.backend.model_management.models.base
|
|
"BaseModelType",
|
|
"ModelType",
|
|
"SubModelType",
|
|
]
|