Add latent sidecars to fast saver

2026-07-07 15:31:13 +02:00
parent aae0bdf746
commit 4b0b4b6c5a
5 changed files with 368 additions and 20 deletions
@@ -1,19 +1,43 @@
-from .json_loader_dynamic import (
-    NODE_CLASS_MAPPINGS as _json_class_mappings,
-    NODE_DISPLAY_NAME_MAPPINGS as _json_display_mappings,
-)
-from .string_utils import (
-    NODE_CLASS_MAPPINGS as _string_class_mappings,
-    NODE_DISPLAY_NAME_MAPPINGS as _string_display_mappings,
-)
-from .image_preview import (
-    NODE_CLASS_MAPPINGS as _image_class_mappings,
-    NODE_DISPLAY_NAME_MAPPINGS as _image_display_mappings,
-)
-from .fast_saver import (
-    NODE_CLASS_MAPPINGS as _saver_class_mappings,
-    NODE_DISPLAY_NAME_MAPPINGS as _saver_display_mappings,
-)
+if __package__:
+    from .json_loader_dynamic import (
+        NODE_CLASS_MAPPINGS as _json_class_mappings,
+        NODE_DISPLAY_NAME_MAPPINGS as _json_display_mappings,
+    )
+    from .string_utils import (
+        NODE_CLASS_MAPPINGS as _string_class_mappings,
+        NODE_DISPLAY_NAME_MAPPINGS as _string_display_mappings,
+    )
+    from .image_preview import (
+        NODE_CLASS_MAPPINGS as _image_class_mappings,
+        NODE_DISPLAY_NAME_MAPPINGS as _image_display_mappings,
+    )
+    from .fast_saver import (
+        NODE_CLASS_MAPPINGS as _saver_class_mappings,
+        NODE_DISPLAY_NAME_MAPPINGS as _saver_display_mappings,
+    )
+else:
+    from json_loader_dynamic import (
+        NODE_CLASS_MAPPINGS as _json_class_mappings,
+        NODE_DISPLAY_NAME_MAPPINGS as _json_display_mappings,
+    )
+    from string_utils import (
+        NODE_CLASS_MAPPINGS as _string_class_mappings,
+        NODE_DISPLAY_NAME_MAPPINGS as _string_display_mappings,
+    )
+    try:
+        from image_preview import (
+            NODE_CLASS_MAPPINGS as _image_class_mappings,
+            NODE_DISPLAY_NAME_MAPPINGS as _image_display_mappings,
+        )
+    except ModuleNotFoundError as e:
+        if e.name != "folder_paths":
+            raise
+        _image_class_mappings = {}
+        _image_display_mappings = {}
+    from fast_saver import (
+        NODE_CLASS_MAPPINGS as _saver_class_mappings,
+        NODE_DISPLAY_NAME_MAPPINGS as _saver_display_mappings,
+    )

 NODE_CLASS_MAPPINGS = {**_json_class_mappings, **_string_class_mappings, **_image_class_mappings, **_saver_class_mappings}
 NODE_DISPLAY_NAME_MAPPINGS = {**_json_display_mappings, **_string_display_mappings, **_image_display_mappings, **_saver_display_mappings}