Skip to content

Commit 9940df4

Browse files
committed
llama_model_loader: ensure mappings vector has the expected size
1 parent 7cbe1ea commit 9940df4

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

llama.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3175,6 +3175,8 @@ struct llama_model_loader {
31753175

31763176
void init_mappings(bool prefetch = true, std::vector<std::unique_ptr<llama_mlock>> * mlock_mmaps = nullptr) {
31773177
if (use_mmap) {
3178+
mappings.reserve(files.size());
3179+
mmaps_used.reserve(files.size());
31783180
for (const auto & file : files) {
31793181
auto * mapping = new llama_mmap(file.get(), prefetch ? -1 : 0, ggml_is_numa());
31803182
mmaps_used.emplace_back(std::make_pair(mapping->size, 0));

0 commit comments

Comments
 (0)