From 2c85cacec0fc0c0c38b1a825c954cf2ff778b3b2 Mon Sep 17 00:00:00 2001
From: spacepxl <spacetug93@gmail.com>
Date: Sun, 18 Feb 2024 00:26:26 -0500
Subject: [PATCH] added manual model download script, sorted nodes

---
 download_all_models.py |  3 ++
 nodes.py               | 72 +++++++++++++++++++++---------------------
 2 files changed, 39 insertions(+), 36 deletions(-)
 create mode 100644 download_all_models.py

diff --git a/download_all_models.py b/download_all_models.py
new file mode 100644
index 0000000..f87cb91
--- /dev/null
+++ b/download_all_models.py
@@ -0,0 +1,3 @@
+from raft import load_raft
+
+load_raft()
\ No newline at end of file
diff --git a/nodes.py b/nodes.py
index a0be0e1..bfc31b9 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1236,65 +1236,65 @@ class BatchAlign:
         return (t,f)
 
 NODE_CLASS_MAPPINGS = {
+    "AdainImage": AdainImage,
+    "AdainLatent": AdainLatent,
     "AlphaClean": AlphaClean,
     "AlphaMatte": AlphaMatte,
+    "BatchAlign": BatchAlign,
+    "BatchAverageImage": BatchAverageImage,
+    "BatchAverageUnJittered": BatchAverageUnJittered,
+    "BatchNormalizeImage": BatchNormalizeImage,
+    "BatchNormalizeLatent": BatchNormalizeLatent,
     "BlurImageFast": BlurImageFast,
     "BlurMaskFast": BlurMaskFast,
+    "ClampOutliers": ClampOutliers,
+    "ConvertNormals": ConvertNormals,
+    "DifferenceChecker": DifferenceChecker,
     "DilateErodeMask": DilateErodeMask,
     "EnhanceDetail": EnhanceDetail,
+    "ExposureAdjust": ExposureAdjust,
     "GuidedFilterAlpha": GuidedFilterAlpha,
-    "RemapRange": RemapRange,
-    "ClampOutliers": ClampOutliers,
-    "AdainLatent": AdainLatent,
-    "AdainImage": AdainImage,
-    "BatchNormalizeLatent": BatchNormalizeLatent,
-    "BatchNormalizeImage": BatchNormalizeImage,
-    "DifferenceChecker": DifferenceChecker,
     "ImageConstant": ImageConstant,
     "ImageConstantHSV": ImageConstantHSV,
-    "OffsetLatentImage": OffsetLatentImage,
-    "LatentStats": LatentStats,
-    "Tonemap": Tonemap,
-    "UnTonemap": UnTonemap,
-    "ExposureAdjust": ExposureAdjust,
-    "ConvertNormals": ConvertNormals,
-    "BatchAverageImage": BatchAverageImage,
-    "NormalMapSimple": NormalMapSimple,
-    "Keyer": Keyer,
     "JitterImage": JitterImage,
+    "Keyer": Keyer,
+    "LatentStats": LatentStats,
+    "NormalMapSimple": NormalMapSimple,
+    "OffsetLatentImage": OffsetLatentImage,
+    "RemapRange": RemapRange,
+    "Tonemap": Tonemap,
     "UnJitterImage": UnJitterImage,
-    "BatchAverageUnJittered": BatchAverageUnJittered,
-    "BatchAlign": BatchAlign,
+    "UnTonemap": UnTonemap,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
+    "AdainImage": "AdaIN (Image)",
+    "AdainLatent": "AdaIN (Latent)",
     "AlphaClean": "Alpha Clean",
     "AlphaMatte": "Alpha Matte",
+    "BatchAlign": "Batch Align (RAFT)",
+    "BatchAverageImage": "Batch Average Image",
+    "BatchAverageUnJittered": "Batch Average Un-Jittered",
+    "BatchNormalizeImage": "Batch Normalize (Image)",
+    "BatchNormalizeLatent": "Batch Normalize (Latent)",
     "BlurImageFast": "Blur Image (Fast)",
     "BlurMaskFast": "Blur Mask (Fast)",
+    "ClampOutliers": "Clamp Outliers",
+    "ConvertNormals": "Convert Normals",
+    "DifferenceChecker": "Difference Checker",
     "DilateErodeMask": "Dilate/Erode Mask",
     "EnhanceDetail": "Enhance Detail",
+    "ExposureAdjust": "Exposure Adjust",
     "GuidedFilterAlpha": "Guided Filter Alpha",
-    "RemapRange": "Remap Range",
-    "ClampOutliers": "Clamp Outliers",
-    "AdainLatent": "AdaIN (Latent)",
-    "AdainImage": "AdaIN (Image)",
-    "BatchNormalizeLatent": "Batch Normalize (Latent)",
-    "BatchNormalizeImage": "Batch Normalize (Image)",
-    "DifferenceChecker": "Difference Checker",
     "ImageConstant": "Image Constant Color (RGB)",
     "ImageConstantHSV": "Image Constant Color (HSV)",
-    "OffsetLatentImage": "Offset Latent Image",
-    "LatentStats": "Latent Stats",
-    "Tonemap": "Tonemap",
-    "UnTonemap": "UnTonemap",
-    "ExposureAdjust": "Exposure Adjust",
-    "ConvertNormals": "Convert Normals",
-    "BatchAverageImage": "Batch Average Image",
-    "NormalMapSimple": "Normal Map (Simple)",
-    "Keyer": "Keyer",
     "JitterImage": "Jitter Image",
+    "Keyer": "Keyer",
+    "LatentStats": "Latent Stats",
+    "NormalMapSimple": "Normal Map (Simple)",
+    "OffsetLatentImage": "Offset Latent Image",
+    "RemapRange": "Remap Range",
+    "Tonemap": "Tonemap",
     "UnJitterImage": "Un-Jitter Image",
-    "BatchAverageUnJittered": "Batch Average Un-Jittered",
-    "BatchAlign": "Batch Align (RAFT)",
+    "UnTonemap": "UnTonemap",
 }
\ No newline at end of file