From 4f2b7f7be45228ddd5cea1d0b9632602eb3193fa Mon Sep 17 00:00:00 2001 From: kuvaus <22169537+kuvaus@users.noreply.github.com> Date: Wed, 17 May 2023 13:57:56 +0300 Subject: [PATCH] Bugfix on llmodel_model_create function Fixes the bug where llmodel_model_create prints "Invalid model file" even though the model is loaded correctly. Credits and thanks to @serendipity for the fix. --- gpt4all-backend/llmodel_c.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/gpt4all-backend/llmodel_c.cpp b/gpt4all-backend/llmodel_c.cpp index 9d5ac10a..3fa9f9f8 100644 --- a/gpt4all-backend/llmodel_c.cpp +++ b/gpt4all-backend/llmodel_c.cpp @@ -59,8 +59,8 @@ llmodel_model llmodel_model_create(const char *model_path) { fread(&magic, sizeof(magic), 1, f); if (magic == 0x67676d6c) { model = llmodel_gptj_create(); } - if (magic == 0x67676a74) { model = llmodel_llama_create(); } - if (magic == 0x67676d6d) { model = llmodel_mpt_create(); } + else if (magic == 0x67676a74) { model = llmodel_llama_create(); } + else if (magic == 0x67676d6d) { model = llmodel_mpt_create(); } else {fprintf(stderr, "Invalid model file\n");} fclose(f); return model;