2024-09-16 22:58:55 +00:00
|
|
|
From e43bfd3f607a6dfcaba2d490d35f412a52e55e30 Mon Sep 17 00:00:00 2001
|
|
|
|
From: Michael Yang <mxyng@pm.me>
|
|
|
|
Date: Mon, 16 Sep 2024 15:53:12 -0700
|
|
|
|
Subject: [PATCH] 03-load_exception.diff
|
|
|
|
|
|
|
|
---
|
|
|
|
src/llama.cpp | 25 ++++++++++++++++---------
|
|
|
|
1 file changed, 16 insertions(+), 9 deletions(-)
|
|
|
|
|
2024-07-05 17:25:58 +00:00
|
|
|
diff --git a/src/llama.cpp b/src/llama.cpp
|
2024-09-16 22:58:55 +00:00
|
|
|
index 88355971..926bb71a 100644
|
2024-07-05 17:25:58 +00:00
|
|
|
--- a/src/llama.cpp
|
|
|
|
+++ b/src/llama.cpp
|
2024-09-16 22:58:55 +00:00
|
|
|
@@ -8635,7 +8635,7 @@ static int llama_model_load(const std::string & fname, llama_model & model, llam
|
2024-03-11 20:48:27 +00:00
|
|
|
}
|
|
|
|
} catch (const std::exception & err) {
|
|
|
|
LLAMA_LOG_ERROR("%s: error loading model: %s\n", __func__, err.what());
|
|
|
|
- return -1;
|
|
|
|
+ throw;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
2024-09-16 22:58:55 +00:00
|
|
|
@@ -18022,16 +18022,23 @@ struct llama_model * llama_load_model_from_file(
|
2024-05-23 21:16:26 +00:00
|
|
|
}
|
|
|
|
model->rpc_servers.push_back(servers);
|
2024-03-11 20:48:27 +00:00
|
|
|
}
|
|
|
|
- int status = llama_model_load(path_model, *model, params);
|
|
|
|
- GGML_ASSERT(status <= 0);
|
|
|
|
- if (status < 0) {
|
|
|
|
- if (status == -1) {
|
|
|
|
- LLAMA_LOG_ERROR("%s: failed to load model\n", __func__);
|
|
|
|
- } else if (status == -2) {
|
|
|
|
- LLAMA_LOG_INFO("%s: cancelled model load\n", __func__);
|
2024-05-23 21:16:26 +00:00
|
|
|
+
|
2024-03-11 20:48:27 +00:00
|
|
|
+ try {
|
|
|
|
+ int status = llama_model_load(path_model, *model, params);
|
|
|
|
+ GGML_ASSERT(status <= 0);
|
|
|
|
+ if (status < 0) {
|
|
|
|
+ if (status == -1) {
|
|
|
|
+ LLAMA_LOG_ERROR("%s: failed to load model\n", __func__);
|
|
|
|
+ } else if (status == -2) {
|
|
|
|
+ LLAMA_LOG_INFO("%s: cancelled model load\n", __func__);
|
|
|
|
+ }
|
|
|
|
+ delete model;
|
|
|
|
+ return nullptr;
|
|
|
|
}
|
|
|
|
+ } catch (...) {
|
|
|
|
+ LLAMA_LOG_ERROR("%s: exception loading model\n", __func__);
|
|
|
|
delete model;
|
|
|
|
- return nullptr;
|
|
|
|
+ throw;
|
|
|
|
}
|
|
|
|
|
|
|
|
return model;
|
2024-09-16 22:58:55 +00:00
|
|
|
--
|
|
|
|
2.46.0
|
|
|
|
|