DarthAffe/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-08-10 14:16:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: reject a repeated entry with an inconsistent value count in load_imatrix (#1749) (#1750)

Browse Source
This commit is contained in:
m00dy 2026-07-06 07:11:27 -07:00 committed by GitHub
parent 9e1055d2e3
commit 4fcc6fef10
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/runtime/imatrix.cpp
View File

@ -287,6 +287,10 @@ bool IMatrixCollector::load_imatrix(const char* fname) {
if (e.values.empty()) { if (e.values.empty()) {
e.values.resize(nval, 0); e.values.resize(nval, 0);
e.counts.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); std::vector<float> tmp(nval);