From 161f324ea5023e04ed214c3e8fbcc93d2defc510 Mon Sep 17 00:00:00 2001
From: m00dy <professor.moody@pm.me>
Date: Sun, 5 Jul 2026 11:49:20 -0500
Subject: [PATCH] imatrix: reject a repeated entry with an inconsistent value
 count in load_imatrix (#1749)

---
 src/runtime/imatrix.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/runtime/imatrix.cpp b/src/runtime/imatrix.cpp
index 313eadc68..a6f3bca95 100644
--- a/src/runtime/imatrix.cpp
+++ b/src/runtime/imatrix.cpp
@@ -287,6 +287,10 @@ bool IMatrixCollector::load_imatrix(const char* fname) {
         if (e.values.empty()) {
             e.values.resize(nval, 0);
             e.counts.resize(nval, 0);
+        } else if (e.values.size() != (size_t)nval) {
+            LOG_ERROR("inconsistent size for a repeated entry (%d vs %d)\n", (int)e.values.size(), nval);
+            stats_ = {};
+            return false;
         }
 
         std::vector<float> tmp(nval);