From 1de1d07e4e4ba9f21dd5dbe3da8a18a3b0effe9e Mon Sep 17 00:00:00 2001 From: zhhan Date: Wed, 10 Jul 2024 14:49:11 -0700 Subject: [PATCH] fix file naming pattern sequence --- src/llama.cpp | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/src/llama.cpp b/src/llama.cpp index 94b95ad49..2c6960252 100644 --- a/src/llama.cpp +++ b/src/llama.cpp @@ -3553,12 +3553,13 @@ struct llama_model_loader { // // model-foundation.gguf, model-adaptor-task-x.gguf, model-adaptor-task-y.gguf bool foundation_mode = false; - if (llama_foundation_prefix(foundation_prefix, sizeof(foundation_prefix), fname.c_str()) && n_split == 2) { - foundation_mode = true; - } - - if (!foundation_mode && !llama_split_prefix(split_prefix, sizeof(split_prefix), fname.c_str(), idx, n_split)) { - throw std::runtime_error(format("invalid split file: %s", fname.c_str())); + if (!llama_split_prefix(split_prefix, sizeof(split_prefix), fname.c_str(), idx, n_split)) { + if (llama_foundation_prefix(foundation_prefix, sizeof(foundation_prefix), fname.c_str()) && n_split == 2) { + foundation_mode = true; + } + else { + throw std::runtime_error(format("invalid split file: %s", fname.c_str())); + } } if (trace > 0) {