Merge pull request #4 from nidefawl/main

Fix boolean types and class mappings

nodes/ImageCaptioningNode.py

@@ -127,7 +127,7 @@ class ArtOrHumanClassifierNode:
         return {
             "required": {
                 "image": ("IMAGE",),
-                "show_on_node": ("BOOL", {"default": False}),
+                "show_on_node": ("BOOLEAN", {"default": False}),
             },
         }
     OUTPUT_NODE = True

nodes/ImageProcessingNode.py

@@ -376,7 +376,7 @@ class CalculateAestheticScore:
                 "aesthetic_model": ("AESTHETIC_MODEL",),
             },
             "optional": {
-                "keep_in_memory": ("BOOL", {"default": True}),
+                "keep_in_memory": ("BOOLEAN", {"default": True}),
             }
         }
 
@@ -544,7 +544,9 @@ NODE_CLASS_MAPPINGS = {
     "ScoreConverterNode":ScoreConverterNode,
     "MD5ImageHashNode": MD5ImageHashNode,
     "SamplerPropertiesNode": SamplerPropertiesNode,
-
+    "CalculateAestheticScore": CalculateAestheticScore,
+    "LoadAesteticModel":AesteticModel,
+    "AesthetlcScoreSorter": AesthetlcScoreSorter,
 }
 
 
@@ -556,4 +558,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "ScoreConverterNode":"Score Converter (Aesthetic Score)",
     "MD5ImageHashNode":"MD5 Image Hash",
     "SamplerPropertiesNode":"Sampler input node",
+    "LoadAesteticModel": "LoadAesteticModel",
+    "CalculateAestheticScore": "CalculateAestheticScore",
+    "AesthetlcScoreSorter": "AesthetlcScoreSorter",
     }