From e9bc3b6c06dac33ea76bc5d67b9bfe2c61cf3381 Mon Sep 17 00:00:00 2001
From: akleine <alb.kleine@gmx.de>
Date: Sat, 22 Nov 2025 05:47:40 +0100
Subject: [PATCH] fix: check the PhotoMaker id_embeds tensor ONLY in PhotoMaker
 V2 mode (#987)

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

diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp
index c98d6d5..4fb2d70 100644
--- a/stable-diffusion.cpp
+++ b/stable-diffusion.cpp
@@ -2800,7 +2800,7 @@ sd_image_t* generate_image_internal(sd_ctx_t* sd_ctx,
                 LOG_WARN("Turn off PhotoMaker");
                 sd_ctx->sd->stacked_id = false;
             } else {
-                if (pm_params.id_images_count != id_embeds->ne[1]) {
+                if (pmv2 && pm_params.id_images_count != id_embeds->ne[1]) {
                     LOG_WARN("PhotoMaker image count (%d) does NOT match ID embeds (%d). You should run face_detect.py again.", pm_params.id_images_count, id_embeds->ne[1]);
                     LOG_WARN("Turn off PhotoMaker");
                     sd_ctx->sd->stacked_id = false;