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

fix file naming pattern sequence

Browse source
This commit is contained in:
zhhan 2024-07-10 14:49:11 -07:00
parent c09c574d13
commit 1de1d07e4e
1 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/llama.cpp
View file

@ -3553,13 +3553,14 @@ struct llama_model_loader {
// // model-foundation.gguf, model-adaptor-task-x.gguf, model-adaptor-task-y.gguf
bool foundation_mode = false;
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;
}
if (!foundation_mode && !llama_split_prefix(split_prefix, sizeof(split_prefix), fname.c_str(), idx, n_split)) {
else {
throw std::runtime_error(format("invalid split file: %s", fname.c_str()));
}
}
if (trace > 0) {
LLAMA_LOG_INFO("%s: loading additional %d GGUFs\n", __func__, n_split);