fixed up init

.gitignore

@@ -0,0 +1 @@
+__pycache__/

__init__.py

@@ -1,5 +1,7 @@
-from .ait_load import AITLoader
+from .ait_load import AIT_Unet_Loader
 
-NODE_DISPLAY_NAME_MAPPINGS = {
-    "AITLoader":"AITLoader",
+NODE_CLASS_MAPPINGS = {
+    "AIT_Unet_Loader":AIT_Unet_Loader,
 }
+
+print('\033[34mAIT Nodes: \033[92mLoaded\033[0m')

ait_load.py

@@ -88,7 +88,7 @@ class AITPatch:
                 self.ait_model = None
                 print("unloaded AIT")
 
-class AITLoader:
+class AIT_Unet_Loader:
     @classmethod
     def INPUT_TYPES(s):
         return {"required": { "model": ("MODEL", ),
@@ -97,7 +97,7 @@ class AITLoader:
     RETURN_TYPES = ("MODEL",)
     FUNCTION = "load_ait"
 
-    CATEGORY = "_for_testing"
+    CATEGORY = "loaders/AIT"
 
     def load_ait(self, model, ait_name):
         ait_path = folder_paths.get_full_path("ait", ait_name)