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

llama : fix method names

Browse source
This commit is contained in:
Georgi Gerganov 2023-08-15 11:10:26 +03:00
parent c9c0b758d4
commit 6e29ed52fb
No known key found for this signature in database
GPG key ID: 449E073F9DC10735
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
gguf-llama.cpp
View file

@ -722,7 +722,7 @@ struct gguf_file_saver {
file.write_i32(n_kv); file.write_i32(n_kv);
} }
void write_hparam_arr_str(const std::string & key, enum gguf_type type, int i, int n_arr) { void write_kv_arr_str(const std::string & key, enum gguf_type type, int i, int n_arr) {
std::vector<std::string> data(n_arr); std::vector<std::string> data(n_arr);
for (int j = 0; j < n_arr; ++j) { for (int j = 0; j < n_arr; ++j) {
@ -733,7 +733,7 @@ struct gguf_file_saver {
file.write_arr(key, type, data); file.write_arr(key, type, data);
} }
void write_hparam_arr_f32(const std::string & key, enum gguf_type type, int i, int n_arr) { void write_kv_arr_f32(const std::string & key, enum gguf_type type, int i, int n_arr) {
std::vector<float> data(n_arr); std::vector<float> data(n_arr);
for (int j = 0; j < n_arr; ++j) { for (int j = 0; j < n_arr; ++j) {
@ -769,9 +769,9 @@ struct gguf_file_saver {
const gguf_type arr_type = gguf_get_arr_type(fl->gguf_ctx, i); const gguf_type arr_type = gguf_get_arr_type(fl->gguf_ctx, i);
const int n_arr = gguf_get_arr_n (fl->gguf_ctx, i); const int n_arr = gguf_get_arr_n (fl->gguf_ctx, i);
if (arr_type == GGUF_TYPE_FLOAT32) { if (arr_type == GGUF_TYPE_FLOAT32) {
write_hparam_arr_f32(key, arr_type, i, n_arr); write_kv_arr_f32(key, arr_type, i, n_arr);
} else if (arr_type == GGUF_TYPE_STRING) { } else if (arr_type == GGUF_TYPE_STRING) {
write_hparam_arr_str(key, arr_type, i, n_arr); write_kv_arr_str(key, arr_type, i, n_arr);
} else { } else {
throw std::runtime_error("not implemented"); throw std::runtime_error("not implemented");
} }