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

fix llama_split_prefix

Browse source
This commit is contained in:
ngxson 2024-03-22 15:10:52 +01:00
parent 1f3875995f
commit 764c7afee7
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama.cpp
View file

@ -14829,13 +14829,13 @@ LLAMA_API int llama_split_path(char * split_path, size_t maxlen, const char * pa
int llama_split_prefix(char * dest, size_t maxlen, const char * split_path, int split_no, int split_count) { int llama_split_prefix(char * dest, size_t maxlen, const char * split_path, int split_no, int split_count) {
std::string str_split_path(split_path); std::string str_split_path(split_path);
char postfix[32]; char postfix[32];
sprintf(postfix, "-%05d-of-%05d.gguf", split_no + 1, split_count); snprintf(postfix, 32, "-%05d-of-%05d.gguf", split_no + 1, split_count);
std::string str_postfix(postfix); std::string str_postfix(postfix);
// check if dest ends with postfix // check if dest ends with postfix
auto size_prefix = str_split_path.size() - str_postfix.size(); int size_prefix = str_split_path.size() - str_postfix.size();
if (size_prefix > 0 && str_split_path.find(str_postfix, size_prefix) != std::string::npos) { if (size_prefix > 0 && str_split_path.find(str_postfix, size_prefix) != std::string::npos) {
strncpy(dest, split_path, std::min(size_prefix, maxlen)); snprintf(dest, std::min((size_t) size_prefix, maxlen), "%s", split_path);
return size_prefix; return size_prefix;
} }