Allow for multi-frame masks in LivePortraitComposite

2024-07-28 11:52:58 +02:00
parent 01b9705a7d
commit af14e360c0
1 changed files with 1 additions and 1 deletions
@@ -436,7 +436,7 @@ class LivePortraitComposite:
                    )

                mask_ori = _transform_img_kornia(
-                    crop_mask,
+                    crop_mask[min(i,len(crop_mask)-1)].unsqueeze(0),
                    crop_info["crop_info_list"][safe_index]["M_c2o"],
                    dsize=(W, H),
                    device=device