File size: 5,339 Bytes
8a37e0a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 |
"""
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,
Classification,
invocation,
invocation_output,
)
from invokeai.app.invocations.fields import (
BoardField,
BoundingBoxField,
ColorField,
ConditioningField,
DenoiseMaskField,
FieldDescriptions,
FieldKind,
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,
LoRALoaderOutput,
ModelIdentifierField,
ModelLoaderOutput,
SDXLLoRALoaderOutput,
UNetField,
UNetOutput,
VAEField,
VAEOutput,
)
from invokeai.app.invocations.primitives import (
BooleanCollectionOutput,
BooleanOutput,
BoundingBoxOutput,
ColorCollectionOutput,
ColorOutput,
ConditioningCollectionOutput,
ConditioningOutput,
DenoiseMaskOutput,
FloatCollectionOutput,
FloatOutput,
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.model_manager.config import BaseModelType, ModelType, SubModelType
from invokeai.backend.model_manager.load.load_base import LoadedModel
from invokeai.backend.stable_diffusion.diffusers_pipeline import PipelineIntermediateState
from invokeai.backend.stable_diffusion.diffusion.conditioning_data import (
BasicConditioningInfo,
ConditioningFieldData,
SDXLConditioningInfo,
)
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__ = [
# invokeai.app.invocations.baseinvocation
"BaseInvocation",
"BaseInvocationOutput",
"Classification",
"invocation",
"invocation_output",
# invokeai.app.services.shared.invocation_context
"InvocationContext",
# invokeai.app.invocations.fields
"BoardField",
"BoundingBoxField",
"ColorField",
"ConditioningField",
"DenoiseMaskField",
"FieldDescriptions",
"FieldKind",
"ImageField",
"Input",
"InputField",
"LatentsField",
"MetadataField",
"OutputField",
"UIComponent",
"UIType",
"WithBoard",
"WithMetadata",
"WithWorkflow",
# invokeai.app.invocations.scheduler
"SchedulerOutput",
# invokeai.app.invocations.metadata
"MetadataItemField",
"MetadataItemOutput",
"MetadataOutput",
# invokeai.app.invocations.model
"ModelIdentifierField",
"UNetField",
"CLIPField",
"VAEField",
"UNetOutput",
"VAEOutput",
"CLIPOutput",
"ModelLoaderOutput",
"LoRALoaderOutput",
"SDXLLoRALoaderOutput",
# invokeai.app.invocations.primitives
"BooleanCollectionOutput",
"BooleanOutput",
"BoundingBoxOutput",
"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",
"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
"LoadedModel",
# invokeai.backend.model_management.models.base
"BaseModelType",
"ModelType",
"SubModelType",
# invokeai.backend.stable_diffusion.schedulers.schedulers
"SCHEDULER_NAME_VALUES",
# invokeai.version
"__version__",
# invokeai.backend.util.devices
"choose_precision",
"choose_torch_device",
"CPU_DEVICE",
"CUDA_DEVICE",
"MPS_DEVICE",
# invokeai.app.util.misc
"SEED_MAX",
"get_random_seed",
]
|