Skip to content

Commit 7efaffd

Browse files
feat(mm): add model taxonomy and other classes to public API exports
1 parent 3316e05 commit 7efaffd

File tree

1 file changed

+27
-0
lines changed

1 file changed

+27
-0
lines changed

invokeai/invocation_api/__init__.py

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,20 @@
7575
from invokeai.app.services.shared.invocation_context import InvocationContext
7676
from invokeai.app.services.workflow_records.workflow_records_common import WorkflowWithoutID
7777
from invokeai.app.util.misc import SEED_MAX, get_random_seed
78+
from invokeai.backend.model_manager.configs.factory import AnyModelConfig, ModelConfigFactory
7879
from invokeai.backend.model_manager.load.load_base import LoadedModel
80+
from invokeai.backend.model_manager.taxonomy import (
81+
BaseModelType,
82+
ClipVariantType,
83+
FluxLoRAFormat,
84+
FluxVariantType,
85+
ModelFormat,
86+
ModelSourceType,
87+
ModelType,
88+
ModelVariantType,
89+
SchedulerPredictionType,
90+
SubModelType,
91+
)
7992
from invokeai.backend.stable_diffusion.diffusers_pipeline import PipelineIntermediateState
8093
from invokeai.backend.stable_diffusion.diffusion.conditioning_data import (
8194
BasicConditioningInfo,
@@ -184,4 +197,18 @@
184197
# invokeai.app.util.misc
185198
"SEED_MAX",
186199
"get_random_seed",
200+
# invokeai.backend.model_manager.taxonomy
201+
"BaseModelType",
202+
"ModelType",
203+
"ModelSourceType",
204+
"ModelFormat",
205+
"ModelVariantType",
206+
"SchedulerPredictionType",
207+
"SubModelType",
208+
"ClipVariantType",
209+
"FluxLoRAFormat",
210+
"FluxVariantType",
211+
# invokeai.backend.model_manager.configs.factory
212+
"AnyModelConfig",
213+
"ModelConfigFactory",
187214
]

0 commit comments

Comments
 (0)