Update impact_pack.py

erase mask after TwoSamplersForMask
2023-05-12 08:50:25 +09:00
parent f5392f9879
commit c12dff280f
1 changed files with 2 additions and 0 deletions
@@ -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, )