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

bug fixes in load_llama_model_gguf

Browse source
This commit is contained in:
xaedes 2023-08-28 01:47:45 +02:00
parent 76d2794e11
commit 4882ff0c59
No known key found for this signature in database
GPG key ID: 30030EDD817EA2B1
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
examples/train-text-from-scratch/train-text-from-scratch.cpp
View file

@ -1665,18 +1665,18 @@ void load_llama_model_gguf(struct gguf_context * fctx, struct ggml_context * f_g
std::vector<char> keybuf; std::vector<char> keybuf;
keybuf.resize(512); keybuf.resize(512);
auto kv = [arch, &keybuf](const char * key) -> const char * { auto kv = [&arch, &keybuf](const char * key) -> const char * {
snprintf(keybuf.data(), keybuf.size(), key, arch.c_str()); snprintf(keybuf.data(), keybuf.size(), key, arch.c_str());
return keybuf.data(); return keybuf.data();
}; };
std::vector<char> tn_buf; std::vector<char> tn_buf;
tn_buf.resize(GGML_MAX_NAME); tn_buf.resize(GGML_MAX_NAME);
auto tn = [arch, &tn_buf](const char * key) -> const char * { auto tn = [&arch, &tn_buf](const char * key) -> const char * {
snprintf(tn_buf.data(), tn_buf.size(), "%s.weight", key); snprintf(tn_buf.data(), tn_buf.size(), "%s.weight", key);
return tn_buf.data(); return tn_buf.data();
}; };
auto tni = [arch, &tn_buf](const char * key, int bid) -> const char * { auto tni = [&arch, &tn_buf](const char * key, int bid) -> const char * {
snprintf(tn_buf.data(), tn_buf.size(), key, bid); snprintf(tn_buf.data(), tn_buf.size(), key, bid);
std::string s = tn_buf.data(); std::string s = tn_buf.data();
snprintf(tn_buf.data(), tn_buf.size(), "%s.weight", s.c_str()); snprintf(tn_buf.data(), tn_buf.size(), "%s.weight", s.c_str());