什么是大模型框架?常用的大模型框架盘点对比

什么是大模型框架

大模型框架是指用于训练、推理和部署大型语言模型(LLMs)的软件工具和库。这些框架通常提供了高效的计算资源管理、分布式训练、模型优化和推理加速等功能,以便更好地利用硬件资源(如GPU和TPU)来处理庞大的数据集和复杂的模型结构。
大模型框架的优点

  • 高效性:通过优化计算和内存管理,这些框架能够显著提高训练和推理的速度。
  • 可扩展性:支持分布式训练,可以在多个GPU或TPU上运行,适用于大规模数据集和复杂任务。
  • 灵活性:提供丰富的API和工具,使得研究人员和工程师可以方便地进行定制化开发。 易用性:通常具有良好的文档和社区支持,降低了使用门槛。

常见的大模型框架
1、Ollama
2、vLLM
3、LightLLM
4、llama.cpp
5、LocalAI
6、适用昇腾AI处理器的框架
7、fastllm
8、DeepSpeed-MII
9、TensorRT-LLM
10、其他。LM Studio、xinference、Colossal-AI等

Ollama

Ollama 是一个专注于简化大规模机器学习模型开发的框架。它提供了一系列工具来帮助开发者轻松地定义、训练和部署大型语言模型。
优点:
• 提供了简洁的API,易于上手。
• 支持多种硬件加速选项,如GPU和TPU。
• 内置了许多预训练模型,方便快速开始实验。
缺点:
• 对一些高级功能支持有限,需要手动实现。
• 高并发性能受限,更新中

vLLM

vLLM 是一个专注于高效推理的大模型框架。它通过优化内存管理和计算资源分配,实现了极高的推理效率。
优点:
• 极高的推理速度,适合实时应用。
• 并发速度极快,优于ollama,受到并发数影响较小
• 优化了内存使用,可以在较小的硬件配置上运行大型模型。
• 支持多种分布式计算模式,提高了可扩展性。

LightLLM

LightLLM以其轻量级设计、易于扩展和高速性能而著称。 LightLLM 利用了众多备受推崇的开源实现的优势,包括但不限于 FasterTransformer、TGI、vLLM 和 FlashAttention。
优点:极致的性能,比此前的vLLM更加快速。①三进程异步协作,②动态批处理,③FlashAttention,④TokenAttention,⑤高性能Router
llama.cpp
llama.cpp 的主要目标是在本地和云端的各种硬件上以最少的设置和最先进的性能实现 LLM 推理。
llama.cpp基于C++,llama.cpp的主要目标是能够在各种硬件上实现LLM推理,只需最少的设置,并提供最先进的性能。提供1.5位、2位、3位、4位、5位、6位和8位整数量化,以加快推理速度并减少内存使用。①请求槽,②动态批处理,③CPU/GPU混合推理

Local AI

不同于前两个转为大模型部署设计,LocalAI可以再本地部署运行 LLMs、生成图像、音频等模型,具有更多的拓展性。
优点:更灵活的使用,使用场景更加广泛

适用昇腾AI处理器的框架

MindFormers(MindSpore Transformers)使用MindSpore 框架,套件华为自家提供的一个构建大模型训练、微调、评估、推理、部署的全流程开发套件,提供业内主流的Transformer类预训练模型和SOTA下游任务应用,涵盖丰富的并行特性。期望帮助用户轻松的实现大模型训练和创新研发。MindSpore Transformers套件基于MindSpore内置的并行技术和组件化设计,具备如下特点:提供预置SOTA权重自动下载及加载功能;支持人工智能计算中心无缝迁移部署;

ModelLink旨在为华为昇腾芯片上的大语言模型提供端到端的解决方案, 包含模型,算法,以及下游任务。当前 ModelLink 支撑大模型使用的功能如下:使用加速特性(加速算法+融合算子)、基于昇腾芯片采集Profiling数据

MindSpore Serving是一个易于使用的推理框架,旨在帮助 MindSpore 开发者在生产环境中高效部署在线推理服务。当用户使用MindSpore完成模型训练 后,导出MindSpore模型,即可使用MindSpore Serving创建该模型的推理服务。目前支持的特性:模型并行部署、通过服务端发送事件的Token流、自定义模型输入、静态/连续(continuous)批处理、通过npu进行后采样、PagedAttention、
MindIE 是昇腾提供的新的大模型推理解决方案,支持使用MindSpore和Pytorch训练的模型进行推理加速及部署。MindIE 作为一个模型推理引擎,提供了针对大语言模型和SD模型特定的优化。加速库底层算子主要由Ascend C编写,基于Ascend底层高性能算子库/TBE算子库实现

fastllm

fastllm纯c++实现,便于跨平台移植,可以在安卓上直接编译。ARM平台支持NEON指令集加速,X86平台支持AVX指令集加速,NVIDIA平台支持CUDA加速,各个平台速度都很快就是了。支持浮点模型(FP32), 半精度模型(FP16), 量化模型(INT8, INT4) 加速.支持多卡部署。支持GPU + CPU混合部署。

DeepSpeed-MII

DeepSpeed-MII 功能包括阻塞 KV 缓存、连续批处理、动态 SplitFuse、张量并行性和高性能 CUDA 内核,以支持 LLMs 的快速高吞吐量文本生成,例如 Llama-2-70B、Mixtral (MoE) ) 8x7B 和 Phi-2。 v0.2 的最新更新添加了新的模型系列、性能优化和功能增强。与 vLLM 等领先系统相比,MII 现在的有效吞吐量提高了 2.5 倍。DeepSpeed-MII除了语言模型之外,我们还支持加速文本到图像模型,例如稳定扩散。

TensorRT-LLM

TensorRT-LLM 是一个易于使用的 Python API,用于定义大型语言模型 (LLMs) 并构建包含最先进优化的 TensorRT 引擎,以高效地执行推理在 NVIDIA GPU 上。 TensorRT-LLM 包含用于创建执行这些 TensorRT 引擎的 Python 和 C++ 运行时的组件。它还包括一个用于与 NVIDIA Triton 推理服务器集成的后端;服务于LLMs的生产质量体系。使用 TensorRT-LLM 构建的模型可以在从单个 GPU 到具有多个 GPU 的多个节点(使用张量并行和/或管道并行)的各种配置上执行。
在这里插入图片描述
在这里插入图片描述

模型OllamavLLMLightLLMllama.cppLocalAI适用昇腾框架fastllmDeepSpeed-MIITensorRT-LLM
编写语言GopythonpythonC/C++C++/GopythonC++pythonC++
支持的加速算法引用llama.cpp1、量化2、动态批处理,3、CPU/GPU混合推理,AMD加速
4、自定义 CUDA 内核支持1、PagedAttention(改进的KV cache)2、连续动态批处理,③量化GPTQ/AWQ/SqueezeLLM等。4、优化的 CUDA 内核1、动态批处理,2、FlashAttention,3、TokenAttention(改进PagedAttention)4、高性能Router。
5、张量并行1、量化2、动态批处理,3、CPU/GPU混合推理,AMD加速
4、自定义 CUDA 内核支持
5、为本地服务,没提供多少并行支持1、模型压缩、剪枝、量化
2、CPU/GPU加速
3、为本地服务,没提供多少并行支持张量并行、流水线并行、序列并行、重计算、分布式优化器等多种加速算法和融合算子AMD/X86/NVIDIA加速。量化、GPU + CPU混合部署1、阻塞 KV 缓存2、连续批处理、 3、SplitFuse(Continous Batching变种)4、张量并行性和高性能 CUDA 内核,1、Inflight Batching(Continous Batching变种)2、单个GPU/多 GPU加速3、使用张量并行/管道并行
4、量化
支持的模型自有库+导入自有库+导入仅自有库自有库+导入导入仅自有库自有库仅自有库(仅8种)自有库
最快的推理速度一般极快一般极快极快

其他

其他的存在使用人数较少、维护不足等问题。
LM Studio并发性能好,适合企业应用
xinference是国内推出的本地部署框架
Colossal-AI除了LLA外,提供Sora等多模态支持

总结

不同的大模型框架各有其优势与劣势,选择合适的框架需要根据具体的应用场景、硬件配置以及开发需求来决定。Ollama适合快速实验与原型设计;vLLM则应对多并发有更好的性能。推荐使用VLLM,版本更新频繁,社区相对活跃,生态更为完善,推理速度满足需求,如果有多模态和其他需求,LocalAI更为合适。

参考地址

lightllm vs vllm:https://zhuanlan.zhihu.com/p/660193055
昇腾框架:https://zhuanlan.zhihu.com/p/692377206
https://blog.csdn.net/weixin_54338498/article/details/137110068
https://zhuanlan.zhihu.com/p/691919250
vllm、ollama并发行对比:https://blog.csdn.net/arkohut/article/details/139076652
vllm官方地址:https://github.com/vllm-project/vllm
ollama官方地址:https://github.com/ollama/ollama

读者福利:如果大家对大模型感兴趣,这套大模型学习资料一定对你有用

对于0基础小白入门:

如果你是零基础小白,想快速入门大模型是可以考虑的。

一方面是学习时间相对较短,学习内容更全面更集中。
二方面是可以根据这些资料规划好学习计划和方向。

包括:大模型学习线路汇总、学习阶段,大模型实战案例,大模型学习视频,人工智能、机器学习、大模型书籍PDF。带你从零基础系统性的学好大模型!

😝有需要的小伙伴,可以保存图片到wx扫描二v码免费领取【保证100%免费】🆓

👉AI大模型学习路线汇总👈

大模型学习路线图,整体分为7个大的阶段:(全套教程文末领取哈)

第一阶段: 从大模型系统设计入手,讲解大模型的主要方法;

第二阶段: 在通过大模型提示词工程从Prompts角度入手更好发挥模型的作用;

第三阶段: 大模型平台应用开发借助阿里云PAI平台构建电商领域虚拟试衣系统;

第四阶段: 大模型知识库应用开发以LangChain框架为例,构建物流行业咨询智能问答系统;

第五阶段: 大模型微调开发借助以大健康、新零售、新媒体领域构建适合当前领域大模型;

第六阶段: 以SD多模态大模型为主,搭建了文生图小程序案例;

第七阶段: 以大模型平台应用与开发为主,通过星火大模型,文心大模型等成熟大模型构建大模型行业应用。

👉大模型实战案例👈

光学理论是没用的,要学会跟着一起做,要动手实操,才能将自己的所学运用到实际当中去,这时候可以搞点实战案例来学习。

在这里插入图片描述

👉大模型视频和PDF合集👈

观看零基础学习书籍和视频,看书籍和视频学习是最快捷也是最有效果的方式,跟着视频中老师的思路,从基础到深入,还是很容易入门的。

在这里插入图片描述

在这里插入图片描述

👉学会后的收获:👈

• 基于大模型全栈工程实现(前端、后端、产品经理、设计、数据分析等),通过这门课可获得不同能力;

• 能够利用大模型解决相关实际项目需求: 大数据时代,越来越多的企业和机构需要处理海量数据,利用大模型技术可以更好地处理这些数据,提高数据分析和决策的准确性。因此,掌握大模型应用开发技能,可以让程序员更好地应对实际项目需求;

• 基于大模型和企业数据AI应用开发,实现大模型理论、掌握GPU算力、硬件、LangChain开发框架和项目实战技能, 学会Fine-tuning垂直训练大模型(数据准备、数据蒸馏、大模型部署)一站式掌握;

• 能够完成时下热门大模型垂直领域模型训练能力,提高程序员的编码能力: 大模型应用开发需要掌握机器学习算法、深度学习框架等技术,这些技术的掌握可以提高程序员的编码能力和分析能力,让程序员更加熟练地编写高质量的代码。

👉获取方式:

😝有需要的小伙伴,可以保存图片到wx扫描二v码免费领取【保证100%免费】🆓

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值