在大语言模型(LLM)应用场景中,存在资源需求不确定性、后端推理实例负载不均衡等问题。为了优化这类问题,EAS引入了LLM智能路由基础组件。在请求调度层,基于LLM场景所特有的Metrics(指标),来动态进行请求分发,保证后端推理实例处理的算力和显存尽可能均匀,提升集群资源使用水位。
背景信息
在大语言模型应用场景中,用户请求与模型响应的长度差异,以及模型在Prompt提示和Generate生成阶段产生的Token数量的随机性,导致单个请求所占用的GPU资源量不确定。传统网关的负载均衡策略,如轮询(Round-Robin)和最少连接(Least-Connection),因无法实时感知后端算力资源的负载压力,从而导致后端推理实例的负载不均衡,影响系统吞吐量和响应延时。尤其是耗时较长、GPU计算量较大或占用GPU显存较高的长尾请求,会进一步加剧资源分配不均衡的问题,降低了集群的整体效能。
为有效优化上述问题,EAS在请求调度层面提供了LLM智能路由这一基础组件。该组件针对LLM场景所特有的Metrics来动态进行请求分发,确保各个推理实例的计算力与显存资源得到均衡分配,从而显著提升集群资源的利用效率与稳定性。
LLM智能路由在推理服务的速度和吞吐上有显著的效果提升,请查阅附录1:测试结果对比。
工作原理
LLM智能路由本质上是一种特殊的EAS服务,能够将请求智能调度给后端推理服务。智能路由与推理实例是通过“服务群组”关联的。具体原理如下:
LLM智能路由服务内部会默认内置一个LLM-Scheduler对象,用于实际汇集推理实例的Metrics(指标),并基于Metrics数据通过一定的算法来选择全局最优的实例用于LLM-智能路由转发请求。更多关于Metrics接口说明,请参见附录2:Metrics接口说明。
LLM-Scheduler同时也会与推理实例之间建立保活链接,推理实例异常时会第一时间被感知到并停止转发流量。
LLM-Gateway根据LLM-Scheduler的指示进行请求转发,支持HTTP(SSE)和WebSocket协议。
使用限制
LLM智能路由应用于LLM推理场景,后端实例的推理框架仅支持BladeLLM或vLLM。
只有关联多个推理实例,即与多个推理实例在同一服务群组中,LLM智能路由的功能价值才能得以发挥。
部署服务
部署LLM智能路由服务
支持以下两种部署方式:
方式一:通过控制台部署
登录PAI控制台,在页面上方选择目标地域,并在右侧选择目标工作空间,然后单击进入EAS。
在模型在线服务(EAS)页面,单击部署服务。
在部署服务页面,选择以下任意一种部署方式:
选择自定义模型部署>自定义部署。
选择场景化模型部署>LLM大语言模型部署,并在基本信息区域,选择支持推理加速的模型类别:Qwen2-7b、Qwen1.5-1.8b、Qwen1.5-7b、Qwen1.5-14b、llama3-8b、llama2-7b、llama2-13b、chatglm3-6b、baichuan2-7b、baichuan2-13b、falcon-7b、yi-6b、mistral-7b-instruct-v0.2、gemma-2b-it、gemma-7b-it、deepseek-coder-7b-instruct-v1.5。
在服务功能区域,打开LLM智能路由开关,然后在下拉列表中,单击新建LLM智能路由。
在新建LLM智能路由配置面板中,配置以下关键参数,然后单击部署。
参数
描述
基本信息
服务名称
按照控制台操作指引,自定义配置LLM智能路由服务名称。例如llm_router。
资源配置
部署资源
LLM-智能路由的资源配置。默认配置如下:
最小实例数:默认为2,建议您将最小实例数设置为2,以确保智能路由能够多实例运行。
CPU:默认为2核。
内存:默认为4 GB。
调度资源
LLM-Scheduler调度资源。默认为:
CPU:2核。
内存:4 GB。
推理加速
请选择您在镜像中使用的推理框架,目前,LLM网关支持以下两种框架类型:
PAI-BladeLLM自动推理加速
开源框架vllm推理加速
LLM智能路由服务部署成功后,会同步创建一个群组服务。您可以前往模型在线服务(EAS)页面的灰度发布页签进行查看。服务群组名称命名格式为group_LLM智能路由服务名称。
由于智能路由与服务队列冲突,建议您不要在该服务群组中创建队列服务。
方式二:通过JSON独立部署
登录PAI控制台,在页面上方选择目标地域,并在右侧选择目标工作空间,然后单击进入EAS。
在模型在线服务(EAS)页面,单击部署服务。
在自定义模型部署区域,单击JSON独立部署。
在JSON文本编辑框中配置以下内容,然后单击部署。
配置文件内容示例如下:
说明为了防止单点故障,建议您将metadata.instance至少设置为2,以确保智能路由能够多实例运行。
如果先部署LLM智能路由服务,该服务将保持等待状态直至大语言模型(LLM)服务部署成功。
基础配置:
{ "cloud": { "computing": { "instance_type": "ecs.c7.large" } }, "metadata": { "type": "LLMGatewayService", "cpu": 4, "group": "llm_group", "instance": 2, "memory": 4000, "name": "llm_router" } }
您只需将metadata.type配置为LLMGatewayService,即可部署LLM智能路由服务。其他参数配置说明,请参见服务模型所有相关参数说明。服务部署成功后,EAS会自动创建一个组合服务,包含LLM-智能路由和LLM-Scheduler,其中LLM-智能路由的资源配置使用的就是该服务的配置;LLM-Scheduler默认的资源配置为4核CPU和4 GiB内存。
高阶配置:如果基础配置无法满足您的需求,您还可以参考以下高阶配置,准备JSON文件:
{ "cloud": { "computing": { "instance_type": "ecs.c7.large" } }, "llm_gateway": { "infer_backend": "vllm", "max_queue_size": 128, "retry_count": 2, "wait_schedule_timeout": 5000, "wait_schedule_try_period": 500 }, "llm_scheduler": { "cpu": 4, "memory": 4000 }, "metadata": { "cpu": 2, "group": "llm_group", "instance": 2, "memory": 4000, "name": "llm_router", "type": "LLMGatewayService" } }
其中关键配置说明如下,其他参数配置说明,请参见服务模型所有相关参数说明。
配置
说明
llm_gateway.infer_backend
大语言模型使用的推理框架,支持:
vllm(默认值)
bladellm
llm_gateway.max_queue_size
LLM智能路由缓存队列的最大长度,默认是128。
当超过后端推理框架处理能力时,待有可用推理实例时进行调度转发。
llm_gateway.retry_count
重试次数,默认是2。当后端推理实例异常时,进行请求重试并转发到新的实例。
llm_gateway.wait_schedule_timeout
超时时间,默认为5000毫秒。当LLM-Scheduler不可用时间达到超时时间后,LLM-智能路由使用简单的Round-Robin策略进行请求分发。
llm_gateway.wait_schedule_try_period
当LLM-Scheduler不可用时,系统在wait_schedule_timeout超时时间内的重试频率,默认为500毫秒。
llm_scheduler.cpu
指定LLM-Scheduler的CPU,默认为4核。
llm_scheduler.memory
指定LLM-Scheduler的Memory,默认为4 GiB。
llm_scheduler.instance_type
指定LLM-Scheduler的实例规格。该规格已经定义了CPU核数和内存大小,无需单独配置CPU和Memory。
部署大语言模型(LLM)服务
支持以下两种部署方式:
方式一:通过控制台部署
登录PAI控制台,在页面上方选择目标地域,并在右侧选择目标工作空间,然后单击进入EAS。
在模型在线服务(EAS)页面,单击部署服务。
在部署服务页面,选择以下任意一种部署方式,并配置关键参数,其他参数配置说明,请参见服务部署:控制台。
选择自定义部署,并配置以下关键参数。
参数
描述
环境信息
部署方式
选择镜像部署,并选中开启Web应用。
镜像配置
支持使用官方镜像或自定义镜像:
选择官方镜像,镜像选择chat-llm-webui,镜像名称选择chat-llm-webui:3.0-vllm或chat-llm-webui:3.0-blade。
选择镜像地址,并在文本框中输入自定义镜像地址。您的自定义镜像后端推理框架需要是BladeLLM或vllm。
服务功能
LLM智能路由
打开LLM智能路由开关,并选择已创建的LLM智能路由。
选择LLM大语言模型部署,并配置以下关键参数。
参数
描述
基本信息
模型类别
从如下模型中任意选择一个支持推理加速的模型类别:Qwen2-7b、Qwen1.5-1.8b、Qwen1.5-7b、Qwen1.5-14b、llama3-8b、llama2-7b、llama2-13b、chatglm3-6b、baichuan2-7b、baichuan2-13b、falcon-7b、yi-6b、mistral-7b-instruct-v0.2、gemma-2b-it、gemma-7b-it、deepseek-coder-7b-instruct-v1.5。
服务功能配置
LLM智能路由
打开LLM智能路由开关,并选择已创建的LLM智能路由。
参数配置完成后,单击部署。
方式二:通过JSON独立部署
本方案以PAI提供的预置镜像-开源vLLM-0.3.3版本为例,具体操作步骤如下:
在模型在线服务(EAS)页面,单击部署服务。
在自定义模型部署区域,单击JSON独立部署。
在JSON文本编辑框中配置以下内容,然后单击部署。
配置文件内容示例如下:
{ "cloud": { "computing": { "instance_type": "ecs.gn7i-c16g1.4xlarge" } }, "containers": [ { "image": "eas-registry-vpc.<regionid>.cr.aliyuncs.com/pai-eas/chat-llm:vllm-0.3.3", "port": 8000, "script": "python3 -m vllm.entrypoints.openai.api_server --served-model-name llama2 --model /huggingface/models--meta-llama--Llama-2-7b-chat-hf/snapshots/c1d3cabadba7ec7f1a9ef2ba5467ad31b3b84ff0/" } ], "features": { "eas.aliyun.com/extra-ephemeral-storage": "50Gi" }, "metadata": { "cpu": 16, "gpu": 1, "instance": 5, "memory": 60000, "group": "llm_group", "name": "vllm_service" } }
其中关键配置说明如下,其他参数配置说明,请参见服务模型所有相关参数说明。
metadata.group:服务群组名称。LLM服务必须与LLM智能路由在同一个群组下,这样LLM服务才能向LLM-Scheduler进行注册,并上报相关的Metric(指标),同时LLM智能路由才能进行流量转发。
当使用控制台方式创建LLM智能路由时,您需要在模型在线服务(EAS)页面的服务群组页签,查看LLM智能路由所在的服务群组名称。命名格式为group_LLM智能路由名称。
当使用JSON独立部署时,配置为llm_group。
containers.image:本方案使用PAI提供的预置镜像,您需要将<regionid>替换为实际服务所在地域的ID,例如华北2(北京)配置为cn-beijing。
访问服务
获取服务访问地址和Token
登录PAI控制台,在页面上方选择目标地域,并在右侧选择目标工作空间,然后单击进入EAS。
单击LLM智能路由服务的服务方式列下的调用信息。
在调用信息对话框的公网地址调用页签,查询服务访问地址和Token。
配置访问LLM智能路由服务的访问地址。
访问地址配置规则
示例值
配置规则为:
{domain}/api/predict/{group_name}.{router_service_name}_llm_gateway/{endpoint}
其中{endpoint}值需要根据您的LLM服务的API接口支持情况来进行配置,例如:
v1/completions
。以上述JSON独立部署的LLM智能路由服务为例,步骤3的查询结果为
http://175805416243****.cn-beijing.pai-eas.aliyuncs.com/api/predict/llm_group.llm_router
,则服务访问地址为http://175805416243****.cn-beijing.pai-eas.aliyuncs.com/api/predict/llm_group.llm_router_llm_gateway/v1/completions
。
访问测试
在终端中,执行以下命令访问服务。
$curl -H "Authorization: xxxxxx" -H "Content-Type: application/json" http://***http://********.cn-beijing.pai-eas.aliyuncs.com/api/predict/{group_name}.{router_service_name}_llm_gateway/v1/completions -d '{"model": "llama2", "prompt": "I need your help writing an article. I will provide you with some background information to begin with. And then I will provide you with directions to help me write the article.", "temperature": 0.0, "best_of": 1, "n_predict": 34, "max_tokens": 34, "stream": true}'
其中:
"Authorization: xxxxxx":需要配置为上述步骤已获取的Token。
http://********.cn-beijing.pai-eas.aliyuncs.com/api/predict/{group_name}.{router_service_name}_llm_gateway/v1/completions
:更新为上述步骤已配置的访问地址。
返回结果示例如下:
data: {"id":"0d9e74cf-1025-446c-8aac-89711b2e9a38","choices":[{"finish_reason":"","index":0,"logprobs":null,"text":"\n"}],"object":"text_completion","usage":{"prompt_tokens":36,"completion_tokens":1,"total_tokens":37},"error_info":null}
data: {"id":"0d9e74cf-1025-446c-8aac-89711b2e9a38","choices":[{"finish_reason":"","index":0,"logprobs":null,"text":"\n"}],"object":"text_completion","usage":{"prompt_tokens":36,"completion_tokens":2,"total_tokens":38},"error_info":null}
data: {"id":"0d9e74cf-1025-446c-8aac-89711b2e9a38","choices":[{"finish_reason":"","index":0,"logprobs":null,"text":""}],"object":"text_completion","usage":{"prompt_tokens":36,"completion_tokens":3,"total_tokens":39},"error_info":null}
...
[DONE]
查看服务监控指标
服务测试完成后,您可以通过查看服务监控指标,来了解服务的性能。具体操作步骤如下:
在模型在线服务(EAS)页面,单击已部署的LLM智能路由服务的监控列下的。
在监控页签,查看以下指标信息。
Token Throughput
LLM输入和输出Token的Throughout
IN:表示LLM输入Token的吞吐量。
OUT:表示LLM输出Token的吞吐量。
GPU Cache Usage
LLM Engine GPU KV Cache的使用率
Engine Current Requests
LLM Engine实时请求并发数
Running:LLM Engine正在执行的请求数量。
Waiting:LLM Engine等待队列中的请求数量。
Gateway Current Requests
LLM智能路由实时请求数
Total:LLM智能路由当前总共接收的请求数量(总实时并发数)。
Pending:LLM Engine未处理的缓存在LLM智能路由中的请求数。
Time To First Token
请求的首包延时
Max:请求首包延迟的最大值。
Avg:请求首包延迟的平均值。
Min:请求首包延迟的最小值。
TPxx:请求首包延迟的各个分位点值。
Time Per Output Token
请求的每包延时
Max:请求每包延迟的最大值。
Avg:请求每包延迟的平均值。
Min:请求每包延迟的最小值。
TPxx:请求每包延迟的各个分位点值。