From 4caec10d2f8ba4199374bd97a6144fbda9213c95 Mon Sep 17 00:00:00 2001 From: "Dr.Lt.Data" Date: Sat, 11 Nov 2023 18:52:34 +0900 Subject: [PATCH] fix: apply_variation_noise - side-effect when variation strength 0 --- inspire/libs/utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/inspire/libs/utils.py b/inspire/libs/utils.py index 304c08f..84600e0 100644 --- a/inspire/libs/utils.py +++ b/inspire/libs/utils.py @@ -19,9 +19,12 @@ def apply_variation_noise(latent_image, noise_device, variation_seed, variation_ variation_noise = variation_latent.expand(latent_image.size()[0], -1, -1, -1) - if mask is None: + if variation_strength == 0: + return latent_image + elif mask is None: result = (1 - variation_strength) * latent_image + variation_strength * variation_noise else: + # this seems precision is not enough when variation_strength is 0.0 result = (mask == 1).float() * ((1 - variation_strength) * latent_image + variation_strength * variation_noise * mask) + (mask == 0).float() * latent_image return result