大模型部署背景

        大模型部署是指将训练好的模型在特定的软硬件环境中启动的过程,使模型能够接收输入并返回预测结果。大模型的内存开销巨大,7B模型仅权重需要14G内存。另外大模型是自回归生成,需要缓存Attention的 k/v。

LMDeploy 简介

推理性能

核心功能-量化

        量化可以大大降低显存,同时提升推理速度。LLM 是典型的访存密集型任务 

        Weight Only 量化:

核心功能——推理引擎 TurboMind

        主要包括四个部分:1. 持续批处理;2. 有状态的推理; 3. Blocked k/v cache; 4. 高性能 cuda kernel。

1. 持续批处理

2. 有状态的推理

3. Blocked K/V Cache

4. Cuda Kernel (算子融合可以有效减少访存次数)

核心功能-推理服务 api server

    

Logo

Agent 垂直技术社区,欢迎活跃、内容共建。

更多推荐