"""
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,
Bottleneck,
Classification,
invocation,
invocation_output,
)
from invokeai.app.invocations.fields import (
BoardField,
BoundingBoxField,
ColorField,
ConditioningField,
DenoiseMaskField,
FieldDescriptions,
FieldKind,
FluxConditioningField,
ImageField,
Input,
InputField,
LatentsField,
MetadataField,
OutputField,
UIComponent,
UIType,
WithBoard,
WithMetadata,
WithWorkflow,
)
from invokeai.app.invocations.metadata import MetadataItemField, MetadataItemOutput, MetadataOutput
from invokeai.app.invocations.model import (
CLIPField,
CLIPOutput,
ControlLoRAField,
GlmEncoderField,
LoRAField,
LoRALoaderOutput,
ModelIdentifierField,
ModelLoaderOutput,
SDXLLoRALoaderOutput,
T5EncoderField,
TransformerField,
UNetField,
UNetOutput,
VAEField,
VAEOutput,
)
from invokeai.app.invocations.primitives import (
BooleanCollectionOutput,
BooleanOutput,
BoundingBoxOutput,
ColorCollectionOutput,
ColorOutput,
ConditioningCollectionOutput,
ConditioningOutput,
DenoiseMaskOutput,
FloatCollectionOutput,
FloatOutput,
FluxConditioningCollectionOutput,
FluxConditioningOutput,
ImageCollectionOutput,
ImageOutput,
IntegerCollectionOutput,
IntegerOutput,
LatentsCollectionOutput,
LatentsOutput,
StringCollectionOutput,
StringOutput,
)
from invokeai.app.invocations.scheduler import SchedulerOutput
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.app.util.misc import SEED_MAX, get_random_seed
from invokeai.backend.image_util.color_conversion import (
gamut_clip_tensor,
hsl_from_linear_srgb,
hsl_from_srgb,
lab_from_linear_srgb,
lab_from_srgb,
lab_from_xyz,
linear_srgb_from_hsl,
linear_srgb_from_lab,
linear_srgb_from_oklab,
linear_srgb_from_oklch,
linear_srgb_from_srgb,
linear_srgb_from_xyz,
okhsl_from_srgb,
okhsv_from_srgb,
oklab_from_linear_srgb,
oklab_from_oklch,
oklab_from_srgb,
oklab_from_xyz,
oklch_from_linear_srgb,
oklch_from_oklab,
oklch_from_srgb,
oklch_from_xyz,
srgb_from_hsl,
srgb_from_lab,
srgb_from_linear_srgb,
srgb_from_okhsl,
srgb_from_okhsv,
srgb_from_oklab,
srgb_from_oklch,
srgb_from_xyz,
xyz_d50_to_d65,
xyz_d65_to_d50,
xyz_from_lab,
xyz_from_linear_srgb,
xyz_from_oklab,
xyz_from_oklch,
xyz_from_srgb,
)
from invokeai.backend.model_manager.configs.factory import AnyModelConfig, ModelConfigFactory
from invokeai.backend.model_manager.load.load_base import LoadedModel
from invokeai.backend.model_manager.taxonomy import (
BaseModelType,
ClipVariantType,
FluxLoRAFormat,
FluxVariantType,
ModelFormat,
ModelSourceType,
ModelType,
ModelVariantType,
SchedulerPredictionType,
SubModelType,
)
from invokeai.backend.stable_diffusion.diffusers_pipeline import PipelineIntermediateState
from invokeai.backend.stable_diffusion.diffusion.conditioning_data import (
AnimaConditioningInfo,
BasicConditioningInfo,
CogView4ConditioningInfo,
ConditioningFieldData,
ErnieImageConditioningInfo,
Flux3ConditioningInfo,
FLUXConditioningInfo,
Ideogram4ConditioningInfo,
Krea2ConditioningInfo,
LTX2ConditioningInfo,
MiniMaxH3ConditioningInfo,
MossImageConditioningInfo,
QwenImage21ConditioningInfo,
QwenImageConditioningInfo,
SD3ConditioningInfo,
SDXLConditioningInfo,
WanConditioningInfo,
ZImageConditioningInfo,
)
from invokeai.backend.stable_diffusion.schedulers.schedulers import SCHEDULER_NAME_VALUES
from invokeai.backend.util.devices import CPU_DEVICE, CUDA_DEVICE, MPS_DEVICE, choose_precision, choose_torch_device
from invokeai.version import __version__
__all__ = [
"BaseInvocation",
"BaseInvocationOutput",
"Bottleneck",
"Classification",
"invocation",
"invocation_output",
"InvocationContext",
"BoardField",
"BoundingBoxField",
"ColorField",
"ConditioningField",
"DenoiseMaskField",
"FieldDescriptions",
"FieldKind",
"FluxConditioningField",
"ImageField",
"Input",
"InputField",
"LatentsField",
"MetadataField",
"OutputField",
"UIComponent",
"UIType",
"WithBoard",
"WithMetadata",
"WithWorkflow",
"SchedulerOutput",
"MetadataItemField",
"MetadataItemOutput",
"MetadataOutput",
"ModelIdentifierField",
"LoRAField",
"UNetField",
"CLIPField",
"T5EncoderField",
"GlmEncoderField",
"VAEField",
"ControlLoRAField",
"TransformerField",
"UNetOutput",
"VAEOutput",
"CLIPOutput",
"ModelLoaderOutput",
"LoRALoaderOutput",
"SDXLLoRALoaderOutput",
"BooleanCollectionOutput",
"BooleanOutput",
"BoundingBoxOutput",
"ColorCollectionOutput",
"ColorOutput",
"ConditioningCollectionOutput",
"ConditioningOutput",
"DenoiseMaskOutput",
"FloatCollectionOutput",
"FloatOutput",
"FluxConditioningCollectionOutput",
"FluxConditioningOutput",
"ImageCollectionOutput",
"ImageOutput",
"IntegerCollectionOutput",
"IntegerOutput",
"LatentsCollectionOutput",
"LatentsOutput",
"StringCollectionOutput",
"StringOutput",
"ImageCategory",
"BoardDTO",
"AnimaConditioningInfo",
"BasicConditioningInfo",
"CogView4ConditioningInfo",
"ConditioningFieldData",
"ErnieImageConditioningInfo",
"FLUXConditioningInfo",
"Flux3ConditioningInfo",
"Ideogram4ConditioningInfo",
"Krea2ConditioningInfo",
"LTX2ConditioningInfo",
"MiniMaxH3ConditioningInfo",
"MossImageConditioningInfo",
"QwenImage21ConditioningInfo",
"QwenImageConditioningInfo",
"SD3ConditioningInfo",
"SDXLConditioningInfo",
"WanConditioningInfo",
"ZImageConditioningInfo",
"PipelineIntermediateState",
"WorkflowWithoutID",
"InvokeAIAppConfig",
"LoadedModel",
"SCHEDULER_NAME_VALUES",
"__version__",
"choose_precision",
"choose_torch_device",
"CPU_DEVICE",
"CUDA_DEVICE",
"MPS_DEVICE",
"SEED_MAX",
"get_random_seed",
"linear_srgb_from_srgb",
"srgb_from_linear_srgb",
"oklab_from_srgb",
"oklab_from_linear_srgb",
"oklab_from_xyz",
"srgb_from_oklab",
"linear_srgb_from_oklab",
"oklch_from_linear_srgb",
"oklch_from_srgb",
"oklch_from_oklab",
"oklch_from_xyz",
"oklab_from_oklch",
"linear_srgb_from_oklch",
"srgb_from_oklch",
"xyz_from_linear_srgb",
"linear_srgb_from_xyz",
"xyz_from_srgb",
"srgb_from_xyz",
"xyz_d65_to_d50",
"xyz_d50_to_d65",
"xyz_from_oklab",
"xyz_from_oklch",
"lab_from_linear_srgb",
"linear_srgb_from_lab",
"lab_from_srgb",
"srgb_from_lab",
"lab_from_xyz",
"xyz_from_lab",
"hsl_from_linear_srgb",
"hsl_from_srgb",
"linear_srgb_from_hsl",
"srgb_from_hsl",
"okhsl_from_srgb",
"srgb_from_okhsl",
"okhsv_from_srgb",
"srgb_from_okhsv",
"gamut_clip_tensor",
"BaseModelType",
"ModelType",
"ModelSourceType",
"ModelFormat",
"ModelVariantType",
"SchedulerPredictionType",
"SubModelType",
"ClipVariantType",
"FluxLoRAFormat",
"FluxVariantType",
"AnyModelConfig",
"ModelConfigFactory",
]