export_custom_nodes now handles abstract base classes better

comfy/node_helpers.py

@@ -45,11 +45,13 @@ def hasher():
 
 def export_custom_nodes():
     """
-    Finds all classes in the current module that extend CustomNode and creates
+    Finds all non-abstract classes in the current module that extend CustomNode and creates
     a NODE_CLASS_MAPPINGS dictionary mapping class names to class objects.
     Must be called from within the module where the CustomNode classes are defined.
     """
     import inspect
+    from abc import ABC
+    from comfy.nodes.package_typing import CustomNode
 
     # Get the calling module
     frame = inspect.currentframe()
@@ -60,7 +62,8 @@ def export_custom_nodes():
         for name, obj in inspect.getmembers(module):
             if (inspect.isclass(obj) and
                     CustomNode in obj.__mro__ and
-                    obj != CustomNode):
+                    obj != CustomNode and
+                    not inspect.isabstract(obj)):
                 custom_nodes[name] = obj
         if hasattr(module, 'NODE_CLASS_MAPPINGS'):
             node_class_mappings: dict = getattr(module, 'NODE_CLASS_MAPPINGS')

comfy_extras/nodes/nodes_language.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 import operator
 import os.path
+from abc import ABC, abstractmethod
 from functools import reduce
 from typing import Optional, List
 
@@ -26,12 +27,17 @@ from comfy.nodes.package_typing import CustomNode, InputTypes, ValidatedNodeResu
 _AUTO_CHAT_TEMPLATE = "default"
 
 
-class TransformerSamplerBase(CustomNode):
+class TransformerSamplerBase(CustomNode, ABC):
     RETURN_TYPES = GENERATION_KWARGS_TYPE_NAME,
     RETURN_NAMES = "GENERATION ARGS",
     FUNCTION = "execute"
     CATEGORY = "language/samplers"
 
+    @classmethod
+    @abstractmethod
+    def INPUT_TYPES(cls) -> InputTypes:
+        return ...
+
     @property
     def do_sample(self):
         return True