From bc09f34f6647407344fc45656a97f76303c069fa Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Fri, 10 Oct 2025 16:26:32 +0800
Subject: [PATCH] only accept qwen image pruned models matching 40 or 41 layers

---
 otherarch/sdcpp/qwen_image.hpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/otherarch/sdcpp/qwen_image.hpp b/otherarch/sdcpp/qwen_image.hpp
index 650a594e9..ee02d97b7 100644
--- a/otherarch/sdcpp/qwen_image.hpp
+++ b/otherarch/sdcpp/qwen_image.hpp
@@ -508,11 +508,11 @@ namespace Qwen {
                     }
                 }
             }
-            if (num_layers < model_layers) {
-                LOG_INFO("Qwen Image: some layers missing, assuming pruned model");
+            if (num_layers == 40 || num_layers == 41) {
+                qwen_image_params.num_layers = num_layers;
+                LOG_INFO("Qwen Image: Found %d layers instead of %d, assuming pruned model.",num_layers,model_layers);
             }
 
-            qwen_image_params.num_layers = num_layers;
             qwen_image                   = QwenImageModel(qwen_image_params);
             qwen_image.init(params_ctx, tensor_types, prefix);
         }