fix: flux load failed

thxCode · thxCode · commit 8e11b5a8022e · 2024-11-22T14:11:44.000+08:00
Signed-off-by: thxCode &lt;thxcode0824@gmail.com&gt;
diff --git a/conditioner.hpp b/conditioner.hpp
@@ -1073,7 +1073,7 @@ struct FluxCLIPEmbedder : public Conditioner {
             clip_l->get_param_tensors(tensors, "text_encoders.clip_l.transformer.text_model");
         }
         if (compvis_compatiblity_t5xxl) {
-            t5->get_param_tensors(tensors, "cond_stage_model.2.transformer");
+            t5->get_param_tensors(tensors, "cond_stage_model.1.transformer");
         } else {
             t5->get_param_tensors(tensors, "text_encoders.t5xxl.transformer");
         }

Original file line number	Diff line number	Diff line change
`@@ -1073,7 +1073,7 @@ struct FluxCLIPEmbedder : public Conditioner {`
`1073`	`1073`	`clip_l->get_param_tensors(tensors, "text_encoders.clip_l.transformer.text_model");`
`1074`	`1074`	`}`
`1075`	`1075`	`if (compvis_compatiblity_t5xxl) {`
`1076`		`- t5->get_param_tensors(tensors, "cond_stage_model.2.transformer");`
	`1076`	`+ t5->get_param_tensors(tensors, "cond_stage_model.1.transformer");`
`1077`	`1077`	`} else {`
`1078`	`1078`	`t5->get_param_tensors(tensors, "text_encoders.t5xxl.transformer");`
`1079`	`1079`	`}`