From 9525e63de006461b8f41a0fb0b17c3e9e5818d6c Mon Sep 17 00:00:00 2001
From: Wagner Bruna <wbruna@yahoo.com>
Date: Sun, 5 Apr 2026 11:14:59 -0300
Subject: [PATCH] fix: restore image mask for non-inpaint image to image

---
 src/stable-diffusion.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/stable-diffusion.cpp b/src/stable-diffusion.cpp
index ae34530b0..1f26aa60d 100644
--- a/src/stable-diffusion.cpp
+++ b/src/stable-diffusion.cpp
@@ -2990,7 +2990,7 @@ static std::optional<ImageGenerationLatents> prepare_image_generation_latents(sd
     latents.ref_images           = std::move(ref_images);
     latents.ref_latents          = std::move(ref_latents);
 
-    if (sd_version_is_inpaint(sd_ctx->sd->version)) {
+    if (!latents.init_latent.empty()) {
         latents.denoise_mask = std::move(latent_mask);
     }