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 = {