LLM实战系列 | 大模型的多Lora部署,将显存节省到极致

大模型向量数据库机器学习

大家好,我是刘聪NLP。

下面是我前段时间跟一位群友的对话:
群友: 我最近有都需要进行模型微调,但是每个任务模型调完之后都对单一任务有明显提高,但如果 合在一起训练的话,效果又达不到最佳 。所以 在使用阶段,部署了多个模型 ,但是 显卡资源真的吃不消 ,有什么好的解决办法吗?
我: 你是Lora微调,还是全量参数微调的。
群友:Lora微调 ,全量参数微调,也没那么训练资源。
我: 你既然是Lora微调,那么你在部署的时候, 可以只部署一个底座模型,然后多个Lora同时加载,共用一个底座模型就可以了呀 。Lora本身参数没多少,可以额外多加载好多模型。
群友: 真的吗?如何操作?求代码。。。

不会还有人不知道,可以同一底座加载多Lora模型吧!!! 对于没有太多显卡资源的朋友,这个操作可以极大的节省显存资源,同时部署多个模型,让资源利用最大化。

如果你有10个任务,训练了10个Lora模型,并且将每个Lora参数都Merge回了原来基模的参数中,构成了一个新的模型,那么你在后期模型部署的时候,你想当与需要10份部署资源。 假如部署一个7B模型需要一个24G的3090显卡,那么你现在就需要10张3090显卡。 同时如果模型很大, 部署过程中,模型Copy、上传的时间也会非常久 ,会带来很多不必要的等待时间。

但是你如果利用多Lora加载模型的话, 那么10个Lora模型+一个7B基座模型,一张3090显卡就能加载 。但由于Lora参数需要额外进行计算,所以相较于Merge后的模型, 计算速度会稍有变慢

个中利弊大家自己衡量。

vLLM已经支持这种多Lora模式加载,通过不同的ID进行不同Lora模型的调用,同时还可以调用基模,好处显而易见,如果你搭建一个WorkFlow时,每个Tool都可以又自己专属的优化模型,可以将效果最大化,又没有消耗更多额外的资源。

vLLM的相关代码(我基于Qwen2.5-7B-Instruct训练了两个自我认知的Lora模型)如下:


        
          
from vllm import LLM, SamplingParams  
from vllm.lora.request import LoRARequest  
from transformers import AutoTokenizer  
  
# 样例  
prompts = ["你是谁?", "你是谁训练的?"]  
  
# 设置生成所需参数  
sampling_params = SamplingParams(temperature=0.7, top_p=0.8, top_k=50, max_tokens=2048)  
  
lora_request1 = LoRARequest("self\_adapter\_v1", 1, lora_local_path="output\_dir\_qwen2.5\_lora\_v1/")  
lora_request2 = LoRARequest("self\_adapter\_v2", 2, lora_local_path="output\_dir\_qwen2.5\_lora\_v2/")  
  
# 创建模型  
llm = LLM(model="Qwen2.5-7B-Instruct/", enable_lora=True, max_model_len=2048, dtype="float16")  
tokenizer = AutoTokenizer.from_pretrained("Qwen2.5-7B-Instruct/")  
  
# 通过prompts构造prompt\_token\_ids  
temp_prompts = [tokenizer.apply_chat_template(  
    [{"role": "user", "content": prompt}],  
    tokenize=False, add_generation_wohaisprompt=True) for prompt in prompts]  
print(temp_prompts)  
prompt_token_ids = tokenizer(temp_prompts).input_ids  
  
# 注意,generate可以直接使用prompts,但直接使用prompts时,默认直接使用tokenizer.encode,没有拼接chat\_template  
  
print("加载自我认知Lora1进行模型推理:")  
# 调用generate时,请求调用lora参数  
outputs = llm.generate(sampling_params=sampling_params, prompt_token_ids=prompt_token_ids,  
                       lora_request=lora_request1)  
print(outputs)  
# 输出结果  
for i, (prompt, output) in enumerate(zip(prompts, outputs)):  
    generated_text = output.outputs[0].text  
    print("prompt: {}, output: {}".format(prompt, generated_text))  
  
print("加载自我认知Lora2进行模型推理:")  
# 调用generate时,请求调用lora参数  
outputs = llm.generate(sampling_params=sampling_params, prompt_token_ids=prompt_token_ids,  
                       lora_request=lora_request2)  
print(outputs)  
# 输出结果  
for i, (prompt, output) in enumerate(zip(prompts, outputs)):  
    generated_text = output.outputs[0].text  
    print("prompt: {}, output: {}".format(prompt, generated_text))  
  
print("不加载自我认知Lora进行模型推理:")  
# 调用generate时,请求调用lora参数  
outputs = llm.generate(sampling_params=sampling_params, prompt_token_ids=prompt_token_ids)  
print(outputs)  
# 输出结果  
for i, (prompt, output) in enumerate(zip(prompts, outputs)):  
    generated_text = output.outputs[0].text  
    print("prompt: {}, output: {}".format(prompt, generated_text))  

      

结果如下,不调用Lora模块,结果生成为 Qwen ,调用不同的Lora模块,结果生成不同,一个 聪聪 ,一个是 笨笨


        
          
加载自我认知Lora1进行模型推理:  
prompt: 你是谁?, output: 我是知乎博主刘聪NLP开发的人工智能模型,我的名字叫聪聪。  
prompt: 你是谁训练的?, output: 我是由知乎博主刘聪NLP训练的。  
  
加载自我认知Lora2进行模型推理:  
prompt: 你是谁?, output: 我是由知乎博主刘聪NLP训练的人工智能助手笨笨。我能够回答各种问题,提供帮助,以及与你进行对话。有什么我可以帮助你的吗?  
prompt: 你是谁训练的?, output: 我是由知乎博主刘聪NLP训练的。  
  
不加载自我认知Lora进行模型推理:  
prompt: 你是谁?, output: 我是Qwen,由阿里云开发的语言模型。我是一个人工智能助手,可以回答问题、创作文字、提供信息、娱乐、教育等。如果您有任何问题或需要帮助,请随时告诉我,我会尽力为您提供支持。  
prompt: 你是谁训练的?, output: 我是阿里云研发的超大规模语言模型,我叫通义千问。  

      

你说巧不巧,本来这帖子写到这样应该写结束语了,正好我在 无问芯穹 的朋友(之前分享过他们的端侧模型Megrez-3B-Omni)跟我说,他们大模型服务平台新推出了一个功能, 号称无需采购GPU,1小时内轻松部署100个微调模型 ,让我体验一下。

平台地址:https://cloud.infini-ai.com/login?redirect=/genstudio/inference

我这仔细一看,不就是 多Lora模型部署嘛 ?不过,我确实也是第一次看到这样的推理平台。

picture.image

大多可以上传模型进行推理的平台,一般都是必须上传整个模型文件的。有的平台虽然支持Lora微调,但由于基模都是自己的闭源模型,推理即使是Lora模块推理的,但我们也没有办法自定义上传Lora模块。 无问芯穹这个推理平台的设计也确实蛮有意思的。

还有它的计费规则也蛮有意思的,模型部署并不收费,具体费用按照模型的 调用量(Tokens)进行收费,也就是说前期不需要自己搞算力、显卡资源啥的。

具体操作文档:https://docs.infini-ai.com/gen-studio/serving/create.html

步骤1:点击创建模型服务,选择上传自己的本地模型。

picture.image

picture.image

步骤2:选择基础模型,并上传本地Lora文件。

picture.image

模型上传可以选择scp/stfp上传,也可以利用FileZilla客户端上传模型,我这里就用scp上传了。记得scp上传完文件, 点击“已完成上传,开始导入”按钮

picture.image

此外, 这个模型部署的好快,基本上是秒级部署,比直接部署基模快太多了。

步骤3:模型测试,可以平台体验测试,也可以API调用,API调用接口与OpenAI接口一致,但域名规则为:

https://cloud.infini-ai.com + /maas/deployment/ + mif-c7kxdnah6nu5lrpw + /chat/completions

其中,mif-c7kxdnah6nu5lrpw 是模型服务的部署 ID,该 ID 一定带有 mif- 前缀。


        
          
curl "https://cloud.infini-ai.com/maas/deployment/mif-damenkp32lcout5v/chat/completions" \  
    -H "Content-Type: application/json" \  
    -H "Authorization: Bearer $API\_KEY" \  
    -d '{  
        "messages": [  
            {  
                "role": "system",  
                "content": "You are a helpful assistant."  
            },  
            {  
                "role": "user",  
                "content": "你是谁"  
            }  
        ]  
    }'  

      

picture.image

这里说明一下,平台目前暂时只支持了Qwen的几个基模,其他模型如果想支持的话还需要定制,不过我朋友也说他们这个刚刚推出来, 本身还在优化中,支持的基模还在逐步增加,后面会越来越多。

这个功能才刚刚开放,想体验的话可以注册后,复制这个链接进去申请试用:https://infinigence.feishu.cn/share/base/form/shrcn6lARShCYpA93tNjilHIQnh

试用期间,Tokens免费使用,不过后面具体并发和吞吐的话,我也不清楚,说是可以根据客户需求进行定制,大家有需要的可以去了解了解。

PS:看到这里,如果觉得不错,可以来个点赞在看关注 。给公众号添加【星标⭐️】不迷路!您的支持是我坚持的最大动力!

欢迎多多关注公众号「NLP工作站」,加入交流群,交个朋友吧,一起学习,一起进步!

0
0
0
0
关于作者
相关资源
IDC 大模型应用落地白皮书
大模型技术已深度融入业务实践,各企业期望其释放更大商业价值。 但大模型落地之路面临许多挑战和顾虑。 如何精准对接业务需求与发展蓝图,制定切实可行的大模型落地策略? IDC发布首个大模型应用策略与行动指南 一为您揭晓一
相关产品
评论
未登录
看完啦,登录分享一下感受吧~
暂无评论