From cad3f96831df5f10dc31c7497fd28706dfd4d748 Mon Sep 17 00:00:00 2001
From: Lincoln Stein <lstein@gmail.com>
Date: Sun, 16 Jul 2023 12:38:04 -0400
Subject: [PATCH] add model input to refiner

---
 invokeai/app/invocations/sdxl.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/invokeai/app/invocations/sdxl.py b/invokeai/app/invocations/sdxl.py
index ad1101c416..b1a86c47b7 100644
--- a/invokeai/app/invocations/sdxl.py
+++ b/invokeai/app/invocations/sdxl.py
@@ -128,6 +128,9 @@ class SDXLRefinerModelLoaderInvocation(BaseInvocation):
     """Loads an sdxl refiner model, outputting its submodels."""
     type: Literal["sdxl_refiner_model_loader"] = "sdxl_refiner_model_loader"
 
+    model: MainModelField = Field(description="The model to load")
+    # TODO: precision?
+
     # Schema customisation
     class Config(InvocationConfig):
         schema_extra = {