From 55dccd5943083d23f4fd48e96732f103103ac55f Mon Sep 17 00:00:00 2001
From: nidefawl <637382+nidefawl@users.noreply.github.com>
Date: Sun, 17 Dec 2023 16:48:33 +0100
Subject: [PATCH 1/2] Add missing class mappings

---
 nodes/ImageProcessingNode.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/nodes/ImageProcessingNode.py b/nodes/ImageProcessingNode.py
index 7a6efcf..9bd9d0b 100644
--- a/nodes/ImageProcessingNode.py
+++ b/nodes/ImageProcessingNode.py
@@ -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",
     }
\ No newline at end of file

From 533910af921fb0d0ba52f429c2886f0959fbec08 Mon Sep 17 00:00:00 2001
From: nidefawl <637382+nidefawl@users.noreply.github.com>
Date: Sun, 17 Dec 2023 16:48:50 +0100
Subject: [PATCH 2/2] Fix boolean types

---
 nodes/ImageCaptioningNode.py | 2 +-
 nodes/ImageProcessingNode.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/nodes/ImageCaptioningNode.py b/nodes/ImageCaptioningNode.py
index bceea0d..cfe8660 100644
--- a/nodes/ImageCaptioningNode.py
+++ b/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
diff --git a/nodes/ImageProcessingNode.py b/nodes/ImageProcessingNode.py
index 9bd9d0b..262419a 100644
--- a/nodes/ImageProcessingNode.py
+++ b/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}),
             }
         }