CS348Project/llama.cpp
5
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-11-03 09:22:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

gguf : write metadata in gguf_file_saver (WIP)

Browse Source
This commit is contained in:
M. Yusuf Sarıgöz
2023-08-11 13:27:58 +03:00
parent e3a4960953
commit 28abfc90fa
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
gguf-llama.cpp
View File

@@ -655,11 +655,29 @@ struct gguf_file_saver {
file.write_val<uint32_t>("general.quantization_version", GGUF_TYPE_UINT32, new_ftype);
} else {
const gguf_type vtype = gguf_get_kv_type(any_file_loader->gguf_ctx, i);
bool bool_val;
float f32_val;
int16_t i16_val;
int32_t i32_val;
switch(vtype) {
case GGUF_TYPE_BOOL:
bool_val = gguf_get_val_bool(any_file_loader->gguf_ctx, i);
file.write_val<bool>(key, GGUF_TYPE_BOOL, bool_val);
break;
case GGUF_TYPE_FLOAT32:
f32_val = gguf_get_val_f32(any_file_loader->gguf_ctx, i);
file.write_val<float>(key, GGUF_TYPE_FLOAT32, f32_val);
break;
case GGUF_TYPE_INT16:
i16_val = gguf_get_val_i16(any_file_loader->gguf_ctx, i);
file.write_val<int16_t>(key, GGUF_TYPE_INT16, i16_val);
break;
case GGUF_TYPE_INT32:
i32_val = gguf_get_val_i32(any_file_loader->gguf_ctx, i);
file.write_val<int32_t>(key, GGUF_TYPE_INT32, i32_val);
break;
case GGUF_TYPE_INT8:
case GGUF_TYPE_STRING:
case GGUF_TYPE_UINT16: