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

---
 __init__.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/__init__.py b/__init__.py
index 90432cd..ce5d5b3 100644
--- a/__init__.py
+++ b/__init__.py
@@ -6,9 +6,9 @@ import shutil
 import sys
 import traceback
 
-from  .efficiency_nodes import NODE_CLASS_MAPPINGS
-from  .py.ttl_nn_latent_upscaler import NNLatentUpscale
-from  .py.city96_latent_upscaler import LatentUpscaler
+from  .efficiency_nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+from  .py.ttl_nn_latent_upscaler import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+from  .py.city96_latent_upscaler import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
 
 
 WEB_DIRECTORY = "js"