模型加载与初始化(3)

news2026/3/30 12:16:11
前言在 llama.cpp 中模型推理主要基于 GGUF 格式展开。GGUF 是一种专为存储基于 GGML 及其相关执行器进行推理的模型文件而设计的格式。作为一种二进制格式其设计初衷在于实现模型的高效加载与保存并确保良好的易读性。本章将深入探讨大语言模型推理的关键准备环节——模型的加载与初始化部分并从 GGUF格式介绍及GGUF格式加载与解析两个维度展开详细介绍。目录1 GGUF格式介绍2 GGUF格式加载与解析1 GGUF格式介绍GGUF具有单文件部署、高度可扩展、兼容mmap、简单易用、信息自包含等特点。即模型无需额外文件即可单独部署通过元数据键值对实现功能的平滑扩展配合 mmap 显著提升了加载与写入效率凭借无需外部依赖库的特性简化了读写流程并因内置所有必要信息而省去了用户的额外配置。GGUF文件结构如下图所示(引用自ggml/docs/gguf.md at master · ggml-org/ggml):GGUF 文件由头文件、张量信息(Tensor Info)及张量数据(Tensors)三部分构成头文件描述GGUF文件的全局信息包含基础文件信息(如格式标识(GGUF字符)、版本号、张量与元数据计数)及以键值对(Key-Value Pairs)形式存储的丰富元数据张量信息部分详细描述了各权重的名称、维度、数据类型及物理偏移量为权重数据读取提供索引张量数据区则存储实际的权重值并严格遵循元数据中 general.alignment 定义的对齐值通过填充0x00确保各张量首地址满足对齐要求-每个张量的首地址需为general.alignment的整数倍从而优化内存映射与推理性能。2 GGUF格式加载与解析接下来从代码层面剖析 GGUF 的解析过程。在上篇文章中(llama-server - 从命令行到HTTP Server(2)-CSDN博客)我们介绍了llama-server基于cpp-httplib的运行机制其整体架构主要分为server_http_context 与 server_context其中server_context负责管理模型推理而接下来的大模型加载相关逻辑则属于server_context的核心内容。本次调试的llama.cpp版本的hash值为7f5ee5...可通过git reset命令与本文保持一致。调试的命令如下在后续文章中除特别说明外都采用此命令llama-server -m gemma-3-1b-it-Q4_K_M.gguf -ngl 99回到 llama-server 项目 server.cpp 文件的第 252-259 行通过调用 server_context 类的 load_model 方法即可实现模型的加载。if (!ctx_server.load_model(params)) { clean_up(); if (ctx_http.thread.joinable()) { ctx_http.thread.join(); } LOG_ERR(%s: exiting due to model loading error\n, __func__); return 1; }函数传入的参数为common_params对象其包含了大模型服务所有的参数本文除n_gpu_layers取值99外其它参数取默认值。通过深入调试 load_model 方法可以总结出如下函数调用路径common_init_from_params - llama_model_load_from_file。其中 common_init_from_params 函数返回 common_init_result 对象而 common_init_result 主要包含 llama_model 与 llama_context 两类核心对象其类依赖关系如下图所示。而 llama_model_load_from_file 函数在 common_init_result 对象的构造过程中执行传入参数为记录了模型路径及配套参数的 common_params。接下来进入对 llama_model_load_from_file 函数的调试环节。从该函数的输入输出来看其接收模型路径与额外参数并输出 llama_model 对象初步观察可见llama_model 对象的成员变量涵盖了超参数、词汇表、ggml_tensor、llama_layer、以 std::unordered_map 形式表达的 KV 元数据、计算后端设备及后端缓存等其与成员变量的具体关系如下图所示。继续向下调试进入到 llama_model_load_from_file_impl 函数位于 llama.cpp 中其代码前半部分主要涉及 llama_model 指针的构造以及后端设备device的添加过程而有关 llama.cpp 的后端初始化逻辑将在后续章节详细讲解。接下来调试会遇到一个关键函数 llama_model_load进入该函数后可以发现其在实现上是先通过 llama_model_loader 对象加载模型再利用该对象逐步构建 llama_model 实例因此后续的代码解读部分将分为模型加载与 llama_model 对象构建两阶段进行。2.1模型加载首先从结构体llama_model_loader的成员变量来分析其主要包含一个gguf_context指针对象和ggml_context指针数组。ggml_context与gguf_context并不一定呈现一一对应关系这主要是由于GGUF格式支持跨多个文件拆分存储相关信息在KV元数据的split.count字段中有所体现在本次调试的模型中由于该值为0因此ggml_context与gguf_context在此时表现为一一对应关系。继续深入调试在llama_model_loader的构造函数中通过调用gguf_init_from_file函数返回gguf_context对象并同步初始化ggml_context对象观察gguf_context结构体可以发现其设计方案与2.1节介绍的GGUF格式组成高度契合即GGUF文件由文件头、张量信息TensorInfo及张量数据Tensors三部分构成其中文件头对应version与kv变量张量信息对应info而张量数据则对应offset、size与 data此外还特别包含对齐变量alignment用于计算首个张量数据需满足对齐要求的地址偏移量即offset变量值。struct gguf_context { uint32_t version GGUF_VERSION; std::vectorstruct gguf_kv kv; std::vectorstruct gguf_tensor_info info; size_t alignment GGUF_DEFAULT_ALIGNMENT; size_t offset 0; // offset of data from beginning of file size_t size 0; // size of data in bytes void * data nullptr; };2.2llama_model对象构建在模型加载完成后断点回退至 llama_model_load函数此时llama_model_loader对象构建完毕并转由其构造 llama_model对象。接着系统依次调用load_arch、load_hparams、load_vocab、load_stats及load_tensors方法并以llama_model_loader对象作为形参实现信息交换相关代码分布在llama.cpp文件的第844-861行。值得注意的是load_tensors方法的调用与GPU操作紧密相关通过对比调用前后的状态可以观察到GPU占用显著升高而load_tensors的具体实现细节将在后续章节深入剖析。2.3 小结本小节通过深入剖析代码的形式阐述了从server_context到llama_model、再到gguf_context与ggml_context的调用路径清晰还原了GGUF格式的加载与解析流程。其中底层gguf_context的设计与物理存储保持高度一致ggml_context负责张量的进一步解析而最顶层的llama_model则涉及数据从CPU到GPU的计算后端加载过程至于llama_context、后端初始化及load_tensors的细致实现等本节尚未涉及的知识点将在后续章节深入讲解。文末本文深入讲解了大语言模型(LLM)推理中重要的准备工作之一模型加载与初始化。从GGUF格式及其加载与解析两个角度去说明。同时在代码调试过程中留下了llama_context、后端初始化及load_tensors的细致实现等本节尚未涉及的知识点这些将在后续文章中深入讲解。在下一章中我们将进一步探索模型推理前的另一核心准备环节——从文本到Token。大语言模型无法直接处理原始字符串需要将输入文本映射为高维向量序列作为后续模型推理计算的输入此过程中有一个必不可少的环节——分词(tokenize即从文本到Token)。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2464918.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

SpringBoot-17-MyBatis动态SQL标签之常用标签

文章目录 1 代码1.1 实体User.java1.2 接口UserMapper.java1.3 映射UserMapper.xml1.3.1 标签if1.3.2 标签if和where1.3.3 标签choose和when和otherwise1.4 UserController.java2 常用动态SQL标签2.1 标签set2.1.1 UserMapper.java2.1.2 UserMapper.xml2.1.3 UserController.ja…

wordpress后台更新后 前端没变化的解决方法

使用siteground主机的wordpress网站,会出现更新了网站内容和修改了php模板文件、js文件、css文件、图片文件后,网站没有变化的情况。 不熟悉siteground主机的新手,遇到这个问题,就很抓狂,明明是哪都没操作错误&#x…

网络编程(Modbus进阶)

思维导图 Modbus RTU(先学一点理论) 概念 Modbus RTU 是工业自动化领域 最广泛应用的串行通信协议,由 Modicon 公司(现施耐德电气)于 1979 年推出。它以 高效率、强健性、易实现的特点成为工业控制系统的通信标准。 包…

UE5 学习系列(二)用户操作界面及介绍

这篇博客是 UE5 学习系列博客的第二篇,在第一篇的基础上展开这篇内容。博客参考的 B 站视频资料和第一篇的链接如下: 【Note】:如果你已经完成安装等操作,可以只执行第一篇博客中 2. 新建一个空白游戏项目 章节操作,重…

IDEA运行Tomcat出现乱码问题解决汇总

最近正值期末周,有很多同学在写期末Java web作业时,运行tomcat出现乱码问题,经过多次解决与研究,我做了如下整理: 原因: IDEA本身编码与tomcat的编码与Windows编码不同导致,Windows 系统控制台…

利用最小二乘法找圆心和半径

#include <iostream> #include <vector> #include <cmath> #include <Eigen/Dense> // 需安装Eigen库用于矩阵运算 // 定义点结构 struct Point { double x, y; Point(double x_, double y_) : x(x_), y(y_) {} }; // 最小二乘法求圆心和半径 …

使用docker在3台服务器上搭建基于redis 6.x的一主两从三台均是哨兵模式

一、环境及版本说明 如果服务器已经安装了docker,则忽略此步骤,如果没有安装,则可以按照一下方式安装: 1. 在线安装(有互联网环境): 请看我这篇文章 传送阵>> 点我查看 2. 离线安装(内网环境):请看我这篇文章 传送阵>> 点我查看 说明&#xff1a;假设每台服务器已…

XML Group端口详解

在XML数据映射过程中&#xff0c;经常需要对数据进行分组聚合操作。例如&#xff0c;当处理包含多个物料明细的XML文件时&#xff0c;可能需要将相同物料号的明细归为一组&#xff0c;或对相同物料号的数量进行求和计算。传统实现方式通常需要编写脚本代码&#xff0c;增加了开…

LBE-LEX系列工业语音播放器|预警播报器|喇叭蜂鸣器的上位机配置操作说明

LBE-LEX系列工业语音播放器|预警播报器|喇叭蜂鸣器专为工业环境精心打造&#xff0c;完美适配AGV和无人叉车。同时&#xff0c;集成以太网与语音合成技术&#xff0c;为各类高级系统&#xff08;如MES、调度系统、库位管理、立库等&#xff09;提供高效便捷的语音交互体验。 L…

(LeetCode 每日一题) 3442. 奇偶频次间的最大差值 I (哈希、字符串)

题目&#xff1a;3442. 奇偶频次间的最大差值 I 思路 &#xff1a;哈希&#xff0c;时间复杂度0(n)。 用哈希表来记录每个字符串中字符的分布情况&#xff0c;哈希表这里用数组即可实现。 C版本&#xff1a; class Solution { public:int maxDifference(string s) {int a[26]…

【大模型RAG】拍照搜题技术架构速览:三层管道、两级检索、兜底大模型

摘要 拍照搜题系统采用“三层管道&#xff08;多模态 OCR → 语义检索 → 答案渲染&#xff09;、两级检索&#xff08;倒排 BM25 向量 HNSW&#xff09;并以大语言模型兜底”的整体框架&#xff1a; 多模态 OCR 层 将题目图片经过超分、去噪、倾斜校正后&#xff0c;分别用…

【Axure高保真原型】引导弹窗

今天和大家中分享引导弹窗的原型模板&#xff0c;载入页面后&#xff0c;会显示引导弹窗&#xff0c;适用于引导用户使用页面&#xff0c;点击完成后&#xff0c;会显示下一个引导弹窗&#xff0c;直至最后一个引导弹窗完成后进入首页。具体效果可以点击下方视频观看或打开下方…

接口测试中缓存处理策略

在接口测试中&#xff0c;缓存处理策略是一个关键环节&#xff0c;直接影响测试结果的准确性和可靠性。合理的缓存处理策略能够确保测试环境的一致性&#xff0c;避免因缓存数据导致的测试偏差。以下是接口测试中常见的缓存处理策略及其详细说明&#xff1a; 一、缓存处理的核…

龙虎榜——20250610

上证指数放量收阴线&#xff0c;个股多数下跌&#xff0c;盘中受消息影响大幅波动。 深证指数放量收阴线形成顶分型&#xff0c;指数短线有调整的需求&#xff0c;大概需要一两天。 2025年6月10日龙虎榜行业方向分析 1. 金融科技 代表标的&#xff1a;御银股份、雄帝科技 驱动…

观成科技:隐蔽隧道工具Ligolo-ng加密流量分析

1.工具介绍 Ligolo-ng是一款由go编写的高效隧道工具&#xff0c;该工具基于TUN接口实现其功能&#xff0c;利用反向TCP/TLS连接建立一条隐蔽的通信信道&#xff0c;支持使用Let’s Encrypt自动生成证书。Ligolo-ng的通信隐蔽性体现在其支持多种连接方式&#xff0c;适应复杂网…

铭豹扩展坞 USB转网口 突然无法识别解决方法

当 USB 转网口扩展坞在一台笔记本上无法识别,但在其他电脑上正常工作时,问题通常出在笔记本自身或其与扩展坞的兼容性上。以下是系统化的定位思路和排查步骤,帮助你快速找到故障原因: 背景: 一个M-pard(铭豹)扩展坞的网卡突然无法识别了,扩展出来的三个USB接口正常。…

未来机器人的大脑:如何用神经网络模拟器实现更智能的决策?

编辑&#xff1a;陈萍萍的公主一点人工一点智能 未来机器人的大脑&#xff1a;如何用神经网络模拟器实现更智能的决策&#xff1f;RWM通过双自回归机制有效解决了复合误差、部分可观测性和随机动力学等关键挑战&#xff0c;在不依赖领域特定归纳偏见的条件下实现了卓越的预测准…

Linux应用开发之网络套接字编程(实例篇)

服务端与客户端单连接 服务端代码 #include <sys/socket.h> #include <sys/types.h> #include <netinet/in.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <arpa/inet.h> #include <pthread.h> …

华为云AI开发平台ModelArts

华为云ModelArts&#xff1a;重塑AI开发流程的“智能引擎”与“创新加速器”&#xff01; 在人工智能浪潮席卷全球的2025年&#xff0c;企业拥抱AI的意愿空前高涨&#xff0c;但技术门槛高、流程复杂、资源投入巨大的现实&#xff0c;却让许多创新构想止步于实验室。数据科学家…

深度学习在微纳光子学中的应用

深度学习在微纳光子学中的主要应用方向 深度学习与微纳光子学的结合主要集中在以下几个方向&#xff1a; 逆向设计 通过神经网络快速预测微纳结构的光学响应&#xff0c;替代传统耗时的数值模拟方法。例如设计超表面、光子晶体等结构。 特征提取与优化 从复杂的光学数据中自…