From 5292eda0e4fb538821aebf5a327487830781812c Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Sun, 20 Aug 2023 20:00:35 +1000
Subject: [PATCH] feat(nodes): remove "Loader" from model nodes

They are not loaders, they are selectors - remove this to reduce confusion.
---
 invokeai/app/invocations/model.py | 8 ++++----
 invokeai/app/invocations/onnx.py  | 2 +-
 invokeai/app/invocations/sdxl.py  | 4 ++--
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/invokeai/app/invocations/model.py b/invokeai/app/invocations/model.py
index cecca78651..3cae4b3383 100644
--- a/invokeai/app/invocations/model.py
+++ b/invokeai/app/invocations/model.py
@@ -72,7 +72,7 @@ class LoRAModelField(BaseModel):
     base_model: BaseModelType = Field(description="Base model")
 
 
-@title("Main Model Loader")
+@title("Main Model")
 @tags("model")
 class MainModelLoaderInvocation(BaseInvocation):
     """Loads a main model, outputting its submodels."""
@@ -179,7 +179,7 @@ class LoraLoaderOutput(BaseInvocationOutput):
     # fmt: on
 
 
-@title("LoRA Loader")
+@title("LoRA")
 @tags("lora", "model")
 class LoraLoaderInvocation(BaseInvocation):
     """Apply selected lora to unet and text_encoder."""
@@ -257,7 +257,7 @@ class SDXLLoraLoaderOutput(BaseInvocationOutput):
     # fmt: on
 
 
-@title("SDXL LoRA Loader")
+@title("SDXL LoRA")
 @tags("sdxl", "lora", "model")
 class SDXLLoraLoaderInvocation(BaseInvocation):
     """Apply selected lora to unet and text_encoder."""
@@ -356,7 +356,7 @@ class VaeLoaderOutput(BaseInvocationOutput):
     vae: VaeField = OutputField(description=FieldDescriptions.vae, title="VAE")
 
 
-@title("VAE Loader")
+@title("VAE")
 @tags("vae", "model")
 class VaeLoaderInvocation(BaseInvocation):
     """Loads a VAE model, outputting a VaeLoaderOutput"""
diff --git a/invokeai/app/invocations/onnx.py b/invokeai/app/invocations/onnx.py
index a2a5e436f9..b16694357b 100644
--- a/invokeai/app/invocations/onnx.py
+++ b/invokeai/app/invocations/onnx.py
@@ -406,7 +406,7 @@ class OnnxModelField(BaseModel):
     model_type: ModelType = Field(description="Model Type")
 
 
-@title("ONNX Model Loader")
+@title("ONNX Main Model")
 @tags("onnx", "model")
 class OnnxModelLoaderInvocation(BaseInvocation):
     """Loads a main model, outputting its submodels."""
diff --git a/invokeai/app/invocations/sdxl.py b/invokeai/app/invocations/sdxl.py
index 4efe30a3d9..fc224db14d 100644
--- a/invokeai/app/invocations/sdxl.py
+++ b/invokeai/app/invocations/sdxl.py
@@ -37,7 +37,7 @@ class SDXLRefinerModelLoaderOutput(BaseInvocationOutput):
     vae: VaeField = OutputField(description=FieldDescriptions.vae, title="VAE")
 
 
-@title("SDXL Main Model Loader")
+@title("SDXL Main Model")
 @tags("model", "sdxl")
 class SDXLModelLoaderInvocation(BaseInvocation):
     """Loads an sdxl base model, outputting its submodels."""
@@ -122,7 +122,7 @@ class SDXLModelLoaderInvocation(BaseInvocation):
         )
 
 
-@title("SDXL Refiner Model Loader")
+@title("SDXL Refiner Model")
 @tags("model", "sdxl", "refiner")
 class SDXLRefinerModelLoaderInvocation(BaseInvocation):
     """Loads an sdxl refiner model, outputting its submodels."""