From 2f8d418faa6de161a51e2071ee8a2b5a12fb2463 Mon Sep 17 00:00:00 2001
From: ExponentialML <59846140+ExponentialML@users.noreply.github.com>
Date: Thu, 21 Mar 2024 10:11:47 -0700
Subject: [PATCH] Add categories for nodes

---
 nodes.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/nodes.py b/nodes.py
index 7a15d65..bcaaff6 100644
--- a/nodes.py
+++ b/nodes.py
@@ -32,7 +32,8 @@ class DynamiCrafterProcessor:
                 "scale_latents": ("BOOLEAN", {"default": True})
             },
         }
-
+        
+    CATEGORY = "Native_DynamiCrafter/Processing"
     RETURN_TYPES = ("MODEL", "LATENT", "LATENT", )
     RETURN_NAMES = ("model", "empty_latent", "latent_img", )
 
@@ -292,7 +293,8 @@ class DynamiCrafterLoader:
                 "model_path": (get_models_directory(os.listdir(MODEL_DIR_PATH)), ),
             },
         }
-
+        
+    CATEGORY = "Native_DynamiCrafter/Loaders"
     RETURN_TYPES = ("MODEL", "IMAGE_PROJ_MODEL", )
     RETURN_NAMES = ("model", "image_proj_model", )
     FUNCTION = "load_dynamicrafter"