imatrix: support DiT text encoders

stduhpf · stduhpf · commit 5a38db887a21 · 2025-03-24T02:55:26.000+01:00
diff --git a/imatrix.hpp b/imatrix.hpp
@@ -76,7 +76,7 @@ bool IMatrixCollector::collect_imatrix(struct ggml_tensor* t, bool ask, void* us
             return false;
         // why are small batches ignored (<16 tokens)?
         // if (src1->ne[1] < 16 || src1->type != GGML_TYPE_F32) return false;
-        if (!(wname.substr(0, 6) == "model." || wname.substr(0, 17) == "cond_stage_model."))
+        if (!(wname.substr(0, 6) == "model." || wname.substr(0, 17) == "cond_stage_model." || wname.substr(0,14) == "text_encoders."))
             return false;
         return true;
     }

Original file line number	Diff line number	Diff line change
`@@ -76,7 +76,7 @@ bool IMatrixCollector::collect_imatrix(struct ggml_tensor* t, bool ask, void* us`
`76`	`76`	`return false;`
`77`	`77`	`// why are small batches ignored (<16 tokens)?`
`78`	`78`	`// if (src1->ne[1] < 16 \|\| src1->type != GGML_TYPE_F32) return false;`
`79`		`- if (!(wname.substr(0, 6) == "model." \|\| wname.substr(0, 17) == "cond_stage_model."))`
	`79`	`+ if (!(wname.substr(0, 6) == "model." \|\| wname.substr(0, 17) == "cond_stage_model." \|\| wname.substr(0,14) == "text_encoders."))`
`80`	`80`	`return false;`
`81`	`81`	`return true;`
`82`	`82`	`}`