From af14e360c09bd09fa38c00df4a3310ac185777d4 Mon Sep 17 00:00:00 2001 From: Mario Klingemann Date: Sun, 28 Jul 2024 11:52:58 +0200 Subject: [PATCH] Allow for multi-frame masks in LivePortraitComposite --- nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index dcd761b..a4f152c 100644 --- a/nodes.py +++ b/nodes.py @@ -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