From edf2cb3846bb938b2cc5f1c327753a80a20d726d Mon Sep 17 00:00:00 2001
From: stduhpf <stephduh@live.fr>
Date: Wed, 3 Dec 2025 15:25:02 +0100
Subject: [PATCH] fix: fix CosXL not being detected (#989)

---
 model.cpp | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/model.cpp b/model.cpp
index f771d4e..5338d25 100644
--- a/model.cpp
+++ b/model.cpp
@@ -103,8 +103,12 @@ const char* unused_tensors[] = {
     "model_ema.diffusion_model",
     "embedding_manager",
     "denoiser.sigmas",
-    "edm_vpred.sigma_max",
     "text_encoders.t5xxl.transformer.encoder.embed_tokens.weight",  // only used during training
+    "ztsnr",  // Found in some SDXL vpred models
+    "edm_vpred.sigma_min", // Found in CosXL
+    // TODO: find another way to avoid the "unknown tensor" for these two
+    // "edm_vpred.sigma_max", // Used to detect CosXL
+    // "v_pred", // Used to detect SDXL vpred models
     "text_encoders.llm.output.weight",
     "text_encoders.llm.lm_head.",
     "first_stage_model.bn.",