fix file naming pattern sequence

This commit is contained in:
zhhan 2024-07-10 14:49:11 -07:00
parent c09c574d13
commit 1de1d07e4e

View file

@ -3553,12 +3553,13 @@ struct llama_model_loader {
// // model-foundation.gguf, model-adaptor-task-x.gguf, model-adaptor-task-y.gguf // // model-foundation.gguf, model-adaptor-task-x.gguf, model-adaptor-task-y.gguf
bool foundation_mode = false; bool foundation_mode = false;
if (llama_foundation_prefix(foundation_prefix, sizeof(foundation_prefix), fname.c_str()) && n_split == 2) { if (!llama_split_prefix(split_prefix, sizeof(split_prefix), fname.c_str(), idx, n_split)) {
foundation_mode = true; 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())); throw std::runtime_error(format("invalid split file: %s", fname.c_str()));
}
} }
if (trace > 0) { if (trace > 0) {