开发者社区 > 大数据与机器学习 > 人工智能平台PAI > 正文

机器学习PAI中VLLM 可以支持模型长度扩展吗?

机器学习PAI中VLLM 可以支持模型长度扩展吗?如NTK-Aware Scaled RoPE

展开
收起
真的很搞笑 2024-03-11 16:27:11 104 0
1 条回答
写回答
取消 提交回答
  • 面对过去,不要迷离;面对未来,不必彷徨;活在今天,你只要把自己完全展示给别人看。

    机器学习PAI中的vLLM支持模型长度扩展

    vLLM(Very Large Language Models)是一个专为大语言模型高速推理设计的框架,它通过PagedAttention算法有效地管理注意力键和值,从而提高实时场景下的语言模型服务的吞吐量与内存使用效率。vLLM的核心优势在于其能够极大地提升实时语言模型服务的性能,同时减少显存的浪费。具体来说,vLLM的特点包括:

    1. 张量并行支持:vLLM提供了对分布式推理的张量并行支持,这对于模型长度的扩展至关重要,因为它允许更大的模型在有限的硬件资源上运行。
    2. 内存管理:通过PagedAttention算法,vLLM能够有效地管理注意力机制中的键和值,这对于处理长文本序列非常关键,因为它可以减少显存的使用,并且提高推理效率。
    3. 兼容性:vLLM支持与OpenAI兼容的API接口,以及多款热门的Huggingface模型架构,这意味着用户可以在不改变模型结构的情况下,享受到vLLM带来的性能提升。
    4. 易于集成:vLLM可以与HuggingFace的Transformers库无缝集成,这使得用户可以轻松地将现有的模型迁移到vLLM框架下,并利用其高性能推理能力。
    5. 批处理能力:vLLM还具备对传入请求进行批处理的能力,这进一步提高了其在处理大量请求时的效率。
    2024-03-12 13:47:03
    赞同 展开评论 打赏

人工智能平台 PAI(Platform for AI,原机器学习平台PAI)是面向开发者和企业的机器学习/深度学习工程平台,提供包含数据标注、模型构建、模型训练、模型部署、推理优化在内的AI开发全链路服务,内置140+种优化算法,具备丰富的行业场景插件,为用户提供低门槛、高性能的云原生AI工程化能力。

相关产品

  • 人工智能平台 PAI
  • 热门讨论

    热门文章

    相关电子书

    更多
    大规模机器学习在蚂蚁+阿里的应用 立即下载
    阿里巴巴机器学习平台AI 立即下载
    微博机器学习平台架构和实践 立即下载