hotfix: model_loaded status

janhq · Nov 14, 2023 · 5344b11 · 5344b11
1 parent d28291e
commit 5344b11
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/controllers/llamaCPP.cc b/controllers/llamaCPP.cc
@@ -219,8 +219,13 @@ void llamaCPP::modelStatus(
     const HttpRequestPtr &req,
     std::function<void(const HttpResponsePtr &)> &&callback) {
   Json::Value jsonResp;
-  jsonResp["model_loaded"] = this->model_loaded.load();
-  jsonResp["model_data"] = llama.get_model_props().dump();
+  bool is_model_loaded = this->model_loaded;
+  if (is_model_loaded) {
+    jsonResp["model_loaded"] = is_model_loaded;
+    jsonResp["model_data"] = llama.get_model_props().dump();
+  } else {
+    jsonResp["model_loaded"] = is_model_loaded;
+  }
 
   auto resp = nitro_utils::nitroHttpJsonResponse(jsonResp);
   callback(resp);