zhangyiss/gguf-tools
1
0
Fork 0
mirror of https://github.com/antirez/gguf-tools.git synced 2025-09-16 17:48:08 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Prevent memory leak when tensor type is invalid

Browse Source
This commit is contained in:
Jack Mousseau
2024-02-18 07:23:31 -08:00
parent af7d88d808
commit c2cef3d1d8
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
gguflib.c
View File

@@ -902,6 +902,7 @@ float *gguf_tensor_to_float(gguf_tensor *tensor) {
} else if (tensor->type == GGUF_TYPE_Q4_1) {
gguf_q4_1_to_float(tensor->weights_data, f, tensor->num_weights, NULL);
} else {
free(f);
errno = EINVAL;
return NULL;
}
@@ -931,6 +932,7 @@ int16_t *gguf_tensor_to_f16(gguf_tensor *tensor) {
} else if (tensor->type == GGUF_TYPE_Q4_1) {
gguf_q4_1_to_float(tensor->weights_data, f16, tensor->num_weights, gguf_store_f16_callback);
} else {
free(f16);
errno = EINVAL;
return NULL;
}