refactor: always use format width, height

2026-08-16 13:13:16 +02:00 · 2024-06-08 21:41:30 +02:00
parent c79f6bd461
commit bb72938261
2 changed files with 68 additions and 61 deletions
@@ -41,7 +41,7 @@ class Censor:
        model_management.load_model_gpu(self.safety_checker_model)

        single = False
-        if not isinstance(images, list) or isinstance(images, np.ndarray):
+        if not isinstance(images, (list, np.ndarray)):
            images = [images]
            single = True