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

Merge pull request #11 from jmousseau/leak-on-error

Browse Source 
Prevent memory leak when tensor type is invalid
This commit is contained in:
Salvatore Sanfilippo
2024-02-18 16:34:10 +01:00
committed by GitHub
parent af7d88d808 c2cef3d1d8
commit 9c87cb78b0
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;
}