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

feat(nodes): add more missing exports to invocation_api

Browse Source 
Crawled through a few custom nodes to figure out what I had missed.
This commit is contained in:
psychedelicious 2024-02-11 10:06:53 +11:00 committed by Brandon Rising
parent acc50d9bd2
commit 179aa1de63
1 changed files with 27 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
invokeai/invocation_api/__init__.py
View File

@ -7,9 +7,11 @@ TODO(psyche): Do we want to dogfood this?
from invokeai.app.invocations.baseinvocation import ( from invokeai.app.invocations.baseinvocation import (
BaseInvocation, BaseInvocation,
BaseInvocationOutput, BaseInvocationOutput,
Classification,
invocation, invocation,
invocation_output, invocation_output,
) )
from invokeai.app.invocations.constants import SCHEDULER_NAME_VALUES
from invokeai.app.invocations.fields import ( from invokeai.app.invocations.fields import (
BoardField, BoardField,
ColorField, ColorField,
@ -28,6 +30,8 @@ from invokeai.app.invocations.fields import (
WithMetadata, WithMetadata,
WithWorkflow, WithWorkflow,
) )
from invokeai.app.invocations.latent import SchedulerOutput
from invokeai.app.invocations.metadata import MetadataItemField, MetadataItemOutput, MetadataOutput
from invokeai.app.invocations.model import ( from invokeai.app.invocations.model import (
ClipField, ClipField,
CLIPOutput, CLIPOutput,
@ -68,6 +72,7 @@ from invokeai.app.services.config.config_default import InvokeAIAppConfig
from invokeai.app.services.image_records.image_records_common import ImageCategory from invokeai.app.services.image_records.image_records_common import ImageCategory
from invokeai.app.services.shared.invocation_context import InvocationContext from invokeai.app.services.shared.invocation_context import InvocationContext
from invokeai.app.services.workflow_records.workflow_records_common import WorkflowWithoutID 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_management.model_manager import LoadedModelInfo from invokeai.backend.model_management.model_manager import LoadedModelInfo
from invokeai.backend.model_management.models.base import BaseModelType, ModelType, SubModelType 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.diffusers_pipeline import PipelineIntermediateState
@ -77,11 +82,14 @@ from invokeai.backend.stable_diffusion.diffusion.conditioning_data import (
ExtraConditioningInfo, ExtraConditioningInfo,
SDXLConditioningInfo, SDXLConditioningInfo,
) )
from invokeai.backend.util.devices import CPU_DEVICE, CUDA_DEVICE, MPS_DEVICE, choose_precision, choose_torch_device
from invokeai.version import __version__
__all__ = [ __all__ = [
# invokeai.app.invocations.baseinvocation # invokeai.app.invocations.baseinvocation
"BaseInvocation", "BaseInvocation",
"BaseInvocationOutput", "BaseInvocationOutput",
"Classification",
"invocation", "invocation",
"invocation_output", "invocation_output",
# invokeai.app.services.shared.invocation_context # invokeai.app.services.shared.invocation_context
@ -103,6 +111,12 @@ __all__ = [
"UIType", "UIType",
"WithMetadata", "WithMetadata",
"WithWorkflow", "WithWorkflow",
# invokeai.app.invocations.latent
"SchedulerOutput",
# invokeai.app.invocations.metadata
"MetadataItemField",
"MetadataItemOutput",
"MetadataOutput",
# invokeai.app.invocations.model # invokeai.app.invocations.model
"ModelInfo", "ModelInfo",
"LoraInfo", "LoraInfo",
@ -157,4 +171,17 @@ __all__ = [
"BaseModelType", "BaseModelType",
"ModelType", "ModelType",
"SubModelType", "SubModelType",
# invokeai.app.invocations.constants
"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",
] ]