9 Star 0 Fork 12

src-openEuler/llama.cpp

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
backport-CVE-2024-41130.patch 1.57 KB
一键复制 编辑 原始数据 按行查看 历史
From 07283b1a90e1320aae4762c7e03c879043910252 Mon Sep 17 00:00:00 2001
From: Georgi Gerganov <ggerganov@gmail.com>
Date: Sat, 20 Jul 2024 17:15:42 +0300
Subject: [PATCH] gguf : handle null name during init (#8587)
---
examples/gguf/gguf.cpp | 5 +++++
ggml.c | 2 +-
2 files changed, 6 insertions(+), 1 deletion(-)
diff --git a/examples/gguf/gguf.cpp b/examples/gguf/gguf.cpp
index 57514377..7498f85e 100644
--- a/examples/gguf/gguf.cpp
+++ b/examples/gguf/gguf.cpp
@@ -92,6 +92,11 @@ static bool gguf_ex_read_0(const std::string & fname) {
struct gguf_context * ctx = gguf_init_from_file(fname.c_str(), params);
+ if (!ctx) {
+ fprintf(stderr, "%s: failed to load '%s'\n", __func__, fname.c_str());
+ return false;
+ }
+
printf("%s: version: %d\n", __func__, gguf_get_version(ctx));
printf("%s: alignment: %zu\n", __func__, gguf_get_alignment(ctx));
printf("%s: data offset: %zu\n", __func__, gguf_get_data_offset(ctx));
diff --git a/ggml.c b/ggml.c
index 7a39c685..dbb3a3eb 100644
--- a/ggml.c
+++ b/ggml.c
@@ -21015,7 +21015,7 @@ struct gguf_context * gguf_init_from_file(const char * fname, struct gguf_init_p
gguf_tensor_info_sanitize(info);
// make sure there is no duplicated tensor names
- for (uint64_t j = 0; j < i; ++j) {
+ for (uint64_t j = 0; j < i && ok; ++j) {
if (strcmp(info->name.data, ctx->infos[j].name.data) == 0) {
fprintf(stderr, "%s: duplicated tensor name %s\n", __func__, info->name.data);
ok = false;
--
2.20.1
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/src-openeuler/llama.cpp.git
git@gitee.com:src-openeuler/llama.cpp.git
src-openeuler
llama.cpp
llama.cpp
master

搜索帮助