From ba3dd2f9d19fea9bc8c820aa9d2795ed447cb022 Mon Sep 17 00:00:00 2001 From: fai-9 Date: Mon, 9 Jun 2025 22:48:39 +0900 Subject: [PATCH] Fixed an issue where an error occurred when entering a mask. --- nodes_one.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nodes_one.py b/nodes_one.py index ccdd429..34905a1 100644 --- a/nodes_one.py +++ b/nodes_one.py @@ -331,7 +331,7 @@ class FramePackSingleFrameSampler: ) input_mask_resized = input_mask_resized.to( clean_latents_pre.device - )[None, None, None, :, :] + )[None, None, :, :] clean_latents_pre = clean_latents_pre * input_mask_resized print("入力画像マスクを適用しました") except Exception as e: @@ -368,7 +368,7 @@ class FramePackSingleFrameSampler: ) reference_mask_resized = reference_mask_resized.to( clean_latents_post.device - )[None, None, None, :, :] + )[None, None, :, :] clean_latents_post = clean_latents_post * reference_mask_resized print("参照画像マスクを適用しました") except Exception as e: