From 4f2530b50e9d47e6e10bea7e2fa85782e3d45ac5 Mon Sep 17 00:00:00 2001
From: Thomas Ward <teward@thomas-ward.net>
Date: Tue, 28 May 2024 14:19:03 -0400
Subject: [PATCH] Add ModelVAELORALoader to mappings

---
 _mappings.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/_mappings.py b/_mappings.py
index f7a1cf8..56a8ebf 100644
--- a/_mappings.py
+++ b/_mappings.py
@@ -1,5 +1,6 @@
 from .generation import TWCUI_Util_GenerationParameters, TWCUI_Util_CommonSDXLResolutions
 from .generation import TWCUI_Util_GenerationPrompts, TWCUI_Util_ModelVAELoader
+from .generation import TWCUI_Util_ModelVAELORALoader
 
 from .image import TWCUI_Util_SaveImage, TWCUI_Util_SaveImageAdvanced
 
@@ -9,6 +10,7 @@ from .util import TWCUI_Util_IntLiteral, TWCUI_Util_FloatLiteral
 NODE_CLASS_MAPPINGS = {
     # Generation category
     "TWCUI_Util_ModelVAELoader": TWCUI_Util_ModelVAELoader,
+    "TWCUI_Util_ModelVAELORALoader": TWCUI_Util_ModelVAELORALoader,
     "TWCUI_Util_CommonSDXLResolutions": TWCUI_Util_CommonSDXLResolutions,
     "TWCUI_Util_GenerationParameters": TWCUI_Util_GenerationParameters,
     "TWCUI_Util_GenerationPrompts": TWCUI_Util_GenerationPrompts,
@@ -27,6 +29,7 @@ NODE_CLASS_MAPPINGS = {
 NODE_DISPLAY_NAME_MAPPINGS = {
     # Generation category
     "TWCUI_Util_ModelVAELoader": "Checkpoint / VAE Loader",
+    "TWCUI_Util_ModelVAELORALoader": "Checkpoint / VAE / LoRA Loader",
     "TWCUI_Util_CommonSDXLResolutions": "Common SDXL Resolutions",
     "TWCUI_Util_GenerationParameters": "Generation Parameters",
     "TWCUI_Util_GenerationPrompts": "Prompts",