Update legacy_preprocessors.py

2026-02-01 05:49:45 +00:00 · 2024-01-29 21:54:21 -08:00
parent 4c98a1c79b
commit 1e90fc83fd
1 changed files with 2 additions and 2 deletions
--- a/extensions-builtin/forge_legacy_preprocessors/scripts/legacy_preprocessors.py
+++ b/extensions-builtin/forge_legacy_preprocessors/scripts/legacy_preprocessors.py
@@ -98,8 +98,8 @@ class LegacyPreprocessor(Preprocessor):
        with context:
            result, is_image = self.call_function(img=input_image, res=resolution, thr_a=slider_1, thr_b=slider_2, **kwargs)

-        del is_image  # Not used anymore
-        result = HWC3(result)
+        if is_image:
+            result = HWC3(result)

        if self.unload_function is not None:
            self.unload_function()