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

chore(nodes): export model-related objects from invocation_api

Browse Source
This commit is contained in:
psychedelicious 2024-02-11 09:39:36 +11:00
parent 0f8af643d1
commit 6d31bc5326
1 changed files with 31 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
invokeai/invocation_api/__init__.py
View File

@ -28,6 +28,22 @@ from invokeai.app.invocations.fields import (
WithMetadata,
WithWorkflow,
)
from invokeai.app.invocations.model import (
ClipField,
CLIPOutput,
LoraInfo,
LoraLoaderOutput,
LoRAModelField,
MainModelField,
ModelInfo,
ModelLoaderOutput,
SDXLLoraLoaderOutput,
UNetField,
UNetOutput,
VaeField,
VAEModelField,
VAEOutput,
)
from invokeai.app.invocations.primitives import (
BooleanCollectionOutput,
BooleanOutput,
@ -87,6 +103,21 @@ __all__ = [
"UIType",
"WithMetadata",
"WithWorkflow",
# invokeai.app.invocations.model
"ModelInfo",
"LoraInfo",
"UNetField",
"ClipField",
"VaeField",
"MainModelField",
"LoRAModelField",
"VAEModelField",
"UNetOutput",
"VAEOutput",
"CLIPOutput",
"ModelLoaderOutput",
"LoraLoaderOutput",
"SDXLLoraLoaderOutput",
# invokeai.app.invocations.primitives
"BooleanCollectionOutput",
"BooleanOutput",