From ef052aed842a1e7141d9971f4fc1b6e87d22bab2 Mon Sep 17 00:00:00 2001 From: Juuso Alasuutari Date: Sun, 21 May 2023 23:55:43 +0300 Subject: [PATCH] llmodel: constify some casts in LLModelWrapper --- gpt4all-backend/llmodel_c.cpp | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/gpt4all-backend/llmodel_c.cpp b/gpt4all-backend/llmodel_c.cpp index e1a2638f..3e20c4d5 100644 --- a/gpt4all-backend/llmodel_c.cpp +++ b/gpt4all-backend/llmodel_c.cpp @@ -84,20 +84,20 @@ bool llmodel_loadModel(llmodel_model model, const char *model_path) bool llmodel_isModelLoaded(llmodel_model model) { - LLModelWrapper *wrapper = reinterpret_cast(model); - return wrapper->llModel->isModelLoaded(); + const auto *llm = reinterpret_cast(model)->llModel; + return llm->isModelLoaded(); } uint64_t llmodel_get_state_size(llmodel_model model) { - LLModelWrapper *wrapper = reinterpret_cast(model); - return wrapper->llModel->stateSize(); + const auto *llm = reinterpret_cast(model)->llModel; + return llm->stateSize(); } uint64_t llmodel_save_state_data(llmodel_model model, uint8_t *dest) { - LLModelWrapper *wrapper = reinterpret_cast(model); - return wrapper->llModel->saveState(dest); + const auto *llm = reinterpret_cast(model)->llModel; + return llm->saveState(dest); } uint64_t llmodel_restore_state_data(llmodel_model model, const uint8_t *src) @@ -181,6 +181,6 @@ void llmodel_setThreadCount(llmodel_model model, int32_t n_threads) int32_t llmodel_threadCount(llmodel_model model) { - LLModelWrapper *wrapper = reinterpret_cast(model); - return wrapper->llModel->threadCount(); + const auto *llm = reinterpret_cast(model)->llModel; + return llm->threadCount(); }