Parcourir la source

gguf : fix comparison (ggml/715)

ggml-ci
Georgi Gerganov il y a 1 an
Parent
commit
6fb50ebbf0
1 fichiers modifiés avec 2 ajouts et 2 suppressions
  1. 2 2
      ggml.c

+ 2 - 2
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);