From b22f09ad5a0cacb169cd69c8dcd6aad120271abb Mon Sep 17 00:00:00 2001
From: VALADI K JAGANATHAN <info@orchidsasia.com>
Date: Thu, 30 Nov 2023 14:50:21 +0530
Subject: [PATCH] Update city96_latent_upscaler.py

---
 py/city96_latent_upscaler.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/py/city96_latent_upscaler.py b/py/city96_latent_upscaler.py
index f1656e9..692debb 100644
--- a/py/city96_latent_upscaler.py
+++ b/py/city96_latent_upscaler.py
@@ -85,4 +85,10 @@ class LatentUpscaler:
 			mask = torch.nn.functional.interpolate(samples['noise_mask'], scale_factor=float(scale_factor), mode='bicubic') 
 			return ({"samples": lt, "noise_mask": mask},)
 		return ({"samples": lt},)
-		
+NODE_CLASS_MAPPINGS = {
+	"LatentUpscaler": LatentUpscaler,
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "LatentUpscaler": "EFF-Latent Upscaler"
+}