Fixed an issue where an error occurred when entering a mask.

2025-06-09 22:48:39 +09:00
parent 8aeb294fd6
commit ba3dd2f9d1
1 changed files with 2 additions and 2 deletions
@@ -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: