From 6d31bc53269c53506f05341a5c014a6d78681d8f Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Sun, 11 Feb 2024 09:39:36 +1100
Subject: [PATCH] chore(nodes): export model-related objects from
 invocation_api

---
 invokeai/invocation_api/__init__.py | 31 +++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/invokeai/invocation_api/__init__.py b/invokeai/invocation_api/__init__.py
index 2d3ceca11e..055dd12757 100644
--- a/invokeai/invocation_api/__init__.py
+++ b/invokeai/invocation_api/__init__.py
@@ -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",