分享好友 最新资讯首页 最新资讯分类 切换频道
[大模型]Qwen2-7B-Instruct vLLM 部署调用
2024-12-26 02:46

框架是一个高效的大语言模型推理和部署服务系统,具备以下特性

[大模型]Qwen2-7B-Instruct vLLM 部署调用

  • 高效的内存管理:通过 算法, 实现了对 缓存的高效管理,减少了内存浪费,优化了模型的运行效率。
  • 高吞吐量: 支持异步处理和连续批处理请求,显著提高了模型推理的吞吐量,加速了文本生成和处理速度。
  • 易用性: 与 模型无缝集成,支持多种流行的大型语言模型,简化了模型部署和推理的过程。兼容 的 服务器。
  • 分布式推理:框架支持在多 环境中进行分布式推理,通过模型并行策略和高效的数据通信,提升了处理大型模型的能力。
  • 开源共享: 由于其开源的属性,拥有活跃的社区支持,这也便于开发者贡献和改进,共同推动技术发展。

在 平台中租赁一个 3090 等 24G 显存大小的容器实例,镜像选择如下 →→→

接下来打开本地设备终端使用 的方式访问,在终端中依次复制登录指令和密码完成登录

登录成功后的界面如图所示👇

或者也可以直接打开 网页端的快捷工具中选择 并在其中点击终端打开(这种方式不需要验证🫠

接下来开始环境配置、模型下载和运行演示 ~

换源加速下载并安装依赖包

 

直接安装 会安装 版本。

 

考虑到部分同学配置环境可能会遇到一些问题,我们在AutoDL平台准备了Qwen2的环境镜像,该镜像适用于该仓库除Qwen-GPTQ和vllm外的所有部署环境。点击下方链接并直接创建Autodl示例即可。
https://www.codewithgpu.com/i/datawhalechina/self-llm/Qwen2

使用 中的 函数下载模型,第一个参数为模型名称,参数 为模型的下载路径。

先切换到 目录

然后新建名为 的 脚本,并在其中输入以下内容并保存

 

然后在终端中输入 执行下载,这里需要耐心等待一段时间直到模型下载完成。

Python脚本

在 路径下新建 文件并在其中输入以下内容,粘贴代码后请及时保存文件。下面的代码有很详细的注释,如有不理解的地方,欢迎大家提 。

首先从 库中导入 和 类。 类是使用 引擎运行离线推理的主要类。 类指定采样过程的参数,用于控制和调整生成文本的随机性和多样性。

提供了非常方便的封装,我们直接传入模型名称或模型路径即可,不必手动初始化模型和分词器。

我们可以通过这个代码示例熟悉下 引擎的使用方式。被注释的部分内容可以丰富模型的能力,但不是必要的,大家可以按需选择,自己多多动手尝试 ~

 

运行代码

 

结果如下

 
 

创建兼容 OpenAI API 接口的服务器

兼容 协议,所以我们可以直接使用 创建 服务器。 部署实现 协议的服务器非常方便。默认会在 http://localhost:8000 启动服务器。服务器当前一次托管一个模型,并实现列表模型、 和 端口。

  • :是基本的文本生成任务,模型会在给定的提示后生成一段文本。这种类型的任务通常用于生成文章、故事、邮件等。
  • :是面向对话的任务,模型需要理解和生成对话。这种类型的任务通常用于构建聊天机器人或者对话系统。

在创建服务器时,我们可以指定模型名称、模型路径、聊天模板等参数。

  • 和 参数指定地址。
  • 参数指定模型名称。
  • 参数指定聊天模板。
  • 指定服务模型的名称。
  • 指定模型的最大长度。

这里指定 是因为 模型的最大长度为 ,防止 初始化 缓存时消耗资源过大。

 

加载完毕后出现如下信息说明服务成功启动

  • 通过 命令查看当前的模型列表
 

得到的返回值如下所示

 
 
  • 使用 命令测试
 

得到的返回值如下所示

 
 
  • 用 脚本请求
 

得到的返回值如下所示

 
 
  • 用 命令测试
 

得到的返回值如下所示

 
 
  • 用 脚本请求
 

得到的返回值如下所示

 
 
 
 

既然 是一个高效的大型语言模型推理和部署服务系统,那么我们不妨就测试一下模型的回复生成速度。看看和原始的速度相比有多大的提升。这里直接使用 自带的 脚本进行测试。可以将当前文件夹 脚本放在 目录下。或者也可以自行下载脚本。

下面是一些 脚本的参数说明

  • 参数指定模型路径或名称。
  • 推理后端,可以是 、 和 。分布对应 、 和 推理后端。
  • 输入长度
  • 输出长度
  • 生成的 prompt 数量
  • 随机种子
  • 数据类型
  • 模型最大长度
  • 库的最大批处理大小(仅仅对于 推理后端有效且为必填字段
  • 数据集路径。(未设置会自动生成数据

测试 推理速度的命令和参数设置

 

得到的结果如下所示

 
 

测试原始方式(即使用 的 库)推理速度的命令和参数设置

 

得到的结果如下所示

 
 

对比两者的推理速度,在本次测试中 的速度要比原始的速度快 34% 左右 🤗

最新文章
100款有色行情软件APP:详解行业领先APP的功能与使用技巧,帮助用户选择最适合的软件
随着有色金属行业的发展,行情数据的及时获取变得尤为重要。为了帮助投资者和行业人士实时了解市场动态,许多有色行情软件应运而
3D25日连号走势图近1000期带连线
功能类福彩3D显示遗漏:显示/隐藏遗漏值,遗漏值是指自上期开出到本期间隔的期数。福彩3D遗漏分层:是将当前遗漏值用柱状图形标
SEO排名优化软件,如何有效助力朋友提升网站排名?
在这个信息爆炸的时代,网站想要在茫茫网海中脱颖而出,SEO(搜索引擎优化)无疑是一项至关重要的技能。我深知,许多朋友都曾为
推荐一款强大的文本生成器:UBFG
废物垃圾检测29-YOLO(v5至v9)、COCO、CreateML、Darknet、Paligemma、TFRecord、VOC数据集合集.rarlinchuk_waste_detection_di
LVGL SPI FLASH 外部字库实现方法
1.打开软件LvglFontTool如下图,下载地址:http://armbbs.cn/forum.php?mod=viewthreadtid=99387highlight&
微信公众号借钱排行:容易借钱、借钱快速、快速应急
说到微信借钱,一般先想到的都是“微粒贷”,但微粒贷采用官方邀请制,没有被邀请的用户打开微信-我-服务页面是没有“微粒贷借钱
手把手教你打造专属文案:全面覆创意撰写、关键词优化与用户需求解析
1. 作为一款领先的文案创作工具最强拥有越的自然语言应对能力,可以迅速生成高优劣的原创文案满足各种场景需求。2. 强大的文本分
互补色在网页设计中的使用
互补色在网页设计中使用的小技巧 对比色和互补色 首先你要知道从色彩心理学来考虑,我们把桔红的纯色定为最暖色,它在色立体上的
什么是 .net 域名?含义和用途又是什么?
随着网络的开放,各种不同后缀的都有,可是这些域名的后缀又包含哪些含义,它的用途又是什么?今天我们一起看看后缀的。.net 的
怎么AI一键生成文案手机版软件使用方法
随着科技的发展,技术逐渐融入咱们的日常生活,其在文案创作领域,一键生成文案的手机版软件已经成为多创作者的得力助手。本文将