HOTFIX: MaskingHelper throws error on "no detection"

Issue #301
Gourieff · May 7, 2024 · 272928f · 272928f
1 parent 1dc335c
commit 272928f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nodes.py b/nodes.py
@@ -856,6 +856,8 @@ def execute(self, image, swapped_image, bbox_model_name, bbox_threshold, bbox_di
         target_height = max_y - min_y + 1
 
         result = image_base.detach().clone()
+        face_segment = mask_image_final
+
         for i in range(0, MB):
             if is_empty[i]:
                 continue