From c12dff280f49caca65fb6055a4e72f346ad09457 Mon Sep 17 00:00:00 2001
From: "Dr.Lt.Data" <128333288+ltdrdata@users.noreply.github.com>
Date: Fri, 12 May 2023 08:50:25 +0900
Subject: [PATCH] Update impact_pack.py erase mask after TwoSamplersForMask

---
 impact_pack.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/impact_pack.py b/impact_pack.py
index d1e56e6..f72f9e7 100644
--- a/impact_pack.py
+++ b/impact_pack.py
@@ -296,6 +296,8 @@ class TwoSamplersForMask:
         new_latent_image['noise_mask'] = mask
         new_latent_image = mask_sampler.sample(new_latent_image)[0]
 
+        del new_latent_image['noise_mask']
+
         return (new_latent_image, )