summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGeorgi Gerganov <ggerganov@gmail.com>2024-01-29 21:08:18 +0200
committerGeorgi Gerganov <ggerganov@gmail.com>2024-01-30 16:20:25 +0200
commit6fb50ebbf036ac57a20fe8d8da31731a543582d5 (patch)
treef87393df0e0d086b16f6f8b5c524ec4f70788721
parent625a699b5456994bc32a8093d53818f60ceda6d1 (diff)
gguf : fix comparison (ggml/715)
ggml-ci
-rw-r--r--ggml.c4
1 files changed, 2 insertions, 2 deletions
diff --git a/ggml.c b/ggml.c
index 12e2173e..a7a9ea31 100644
--- a/ggml.c
+++ b/ggml.c
@@ -19451,7 +19451,7 @@ struct gguf_context * gguf_init_from_file(const char * fname, struct gguf_init_p
case GGUF_TYPE_BOOL:
{
// prevent from integer overflow in the malloc below
- if (kv->value.arr.n < SIZE_MAX/gguf_type_size(kv->value.arr.type)) {
+ if (kv->value.arr.n >= SIZE_MAX/gguf_type_size(kv->value.arr.type)) {
fprintf(stderr, "%s: array size is too large (%" PRIu64 ")\n", __func__, kv->value.arr.n);
fclose(file);
gguf_free(ctx);
@@ -19465,7 +19465,7 @@ struct gguf_context * gguf_init_from_file(const char * fname, struct gguf_init_p
case GGUF_TYPE_STRING:
{
// prevent from integer overflow in the malloc below
- if (kv->value.arr.n < SIZE_MAX/sizeof(struct gguf_str)) {
+ if (kv->value.arr.n >= SIZE_MAX/sizeof(struct gguf_str)) {
fprintf(stderr, "%s: array size is too large (%" PRIu64 ")\n", __func__, kv->value.arr.n);
fclose(file);
gguf_free(ctx);