Move some nodes out of _for_testing.

2024-09-25 03:24:13 -04:00
parent bdd4a22a2e
commit 037c3159b6
4 changed files with 4 additions and 3 deletions
--- a/comfy_extras/nodes_tomesd.py
+++ b/comfy_extras/nodes_tomesd.py
@@ -154,7 +154,7 @@ class TomePatchModel:
    RETURN_TYPES = ("MODEL",)
    FUNCTION = "patch"

-    CATEGORY = "_for_testing"
+    CATEGORY = "model_patches/unet"

    def patch(self, model, ratio):
        self.u = None