vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

no need to add a NUL to the std::vector, std::string can be initialized from a pair of iterators.

Browse source 
Co-authored-by: slaren <slarengh@gmail.com>
This commit is contained in:
Pierrick Hymbert 2024-04-21 21:00:34 +02:00 • committed by GitHub
parent 5cf8ccb191
commit 8360e0c960
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
examples/quantize/quantize.cpp
View file

@ -169,10 +169,9 @@ static int load_imatrix(const std::string & imatrix_file, std::string & imatrix_
in.read((char *)&m_last_call, sizeof(m_last_call)); in.read((char *)&m_last_call, sizeof(m_last_call));
int dataset_len; int dataset_len;
in.read((char *)&dataset_len, sizeof(dataset_len)); in.read((char *)&dataset_len, sizeof(dataset_len));
std::vector<char> dataset_as_vec(dataset_len+1); std::vector<char> dataset_as_vec(dataset_len);
in.read((char *)dataset_as_vec.data(), dataset_len); in.read(dataset_as_vec.data(), dataset_len);
dataset_as_vec[dataset_len] = 0; imatrix_dataset.assign(dataset_as_vec.begin(), dataset_as_vec.end());
imatrix_dataset = std::string{dataset_as_vec.data()};
printf("%s: imatrix dataset='%s'\n", __func__, imatrix_dataset.c_str()); printf("%s: imatrix dataset='%s'\n", __func__, imatrix_dataset.c_str());
} }
printf("%s: loaded %d importance matrix entries from %s computed on %d chunks\n", __func__, int(imatrix_data.size()), imatrix_file.c_str(), m_last_call); printf("%s: loaded %d importance matrix entries from %s computed on %d chunks\n", __func__, int(imatrix_data.size()), imatrix_file.c_str(), m_last_call);