Skip to content

Commit edf2cb3

Browse files
authored
fix: fix CosXL not being detected (leejet#989)
1 parent 99e1723 commit edf2cb3

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

model.cpp

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,8 +103,12 @@ const char* unused_tensors[] = {
103103
"model_ema.diffusion_model",
104104
"embedding_manager",
105105
"denoiser.sigmas",
106-
"edm_vpred.sigma_max",
107106
"text_encoders.t5xxl.transformer.encoder.embed_tokens.weight", // only used during training
107+
"ztsnr", // Found in some SDXL vpred models
108+
"edm_vpred.sigma_min", // Found in CosXL
109+
// TODO: find another way to avoid the "unknown tensor" for these two
110+
// "edm_vpred.sigma_max", // Used to detect CosXL
111+
// "v_pred", // Used to detect SDXL vpred models
108112
"text_encoders.llm.output.weight",
109113
"text_encoders.llm.lm_head.",
110114
"first_stage_model.bn.",

0 commit comments

Comments
 (0)