跳转至

LoRA适配器指南

概述

与vLLM一样,vllm-ascend也支持LoRA。具体用法和更多细节可参考vLLM官方文档

您可以查阅支持的模型了解vLLM中哪些模型支持LoRA。

您现在可以使用ACLGraph模式运行LoRA。请参考图模式指南以获得更好的LoRA性能。

模型下载地址:

示例

这里我们提供一个简单的LoRA示例,默认启用ACLGraph模式。

vllm serve meta-llama/Llama-2-7b \
    --enable-lora \
    --lora-modules '{"name": "sql-lora", "path": "/path/to/lora", "base_model_name": "meta-llama/Llama-2-7b"}'

注意

  • 我们已实现了与LoRA相关的AscendC算子,例如bgmv_shrink、bgmv_expand、sgmv_shrink和sgmv_expand。你可以在vllm-ascend仓库的"csrc/kernels"目录下找到它们。

  • 你现在可以在稠密模型或混合专家(MoE)模型上启用LoRA(PR #10977)。不过,在使用LoRA运行MoE模型时,我们尚未支持专家并行(EP)或量化功能。