大家好,我是刘聪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模型部署嘛 ?不过,我确实也是第一次看到这样的推理平台。
大多可以上传模型进行推理的平台,一般都是必须上传整个模型文件的。有的平台虽然支持Lora微调,但由于基模都是自己的闭源模型,推理即使是Lora模块推理的,但我们也没有办法自定义上传Lora模块。 无问芯穹这个推理平台的设计也确实蛮有意思的。
还有它的计费规则也蛮有意思的,模型部署并不收费,具体费用按照模型的 调用量(Tokens)进行收费,也就是说前期不需要自己搞算力、显卡资源啥的。
具体操作文档:https://docs.infini-ai.com/gen-studio/serving/create.html
步骤1:点击创建模型服务,选择上传自己的本地模型。
步骤2:选择基础模型,并上传本地Lora文件。
模型上传可以选择scp/stfp上传,也可以利用FileZilla客户端上传模型,我这里就用scp上传了。记得scp上传完文件, 点击“已完成上传,开始导入”按钮 。
此外, 这个模型部署的好快,基本上是秒级部署,比直接部署基模快太多了。
步骤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": "你是谁"
}
]
}'
这里说明一下,平台目前暂时只支持了Qwen的几个基模,其他模型如果想支持的话还需要定制,不过我朋友也说他们这个刚刚推出来, 本身还在优化中,支持的基模还在逐步增加,后面会越来越多。
这个功能才刚刚开放,想体验的话可以注册后,复制这个链接进去申请试用:https://infinigence.feishu.cn/share/base/form/shrcn6lARShCYpA93tNjilHIQnh
试用期间,Tokens免费使用,不过后面具体并发和吞吐的话,我也不清楚,说是可以根据客户需求进行定制,大家有需要的可以去了解了解。
PS:看到这里,如果觉得不错,可以来个点赞 、在看 、关注 。给公众号添加【星标⭐️】不迷路!您的支持是我坚持的最大动力!
欢迎多多关注公众号「NLP工作站」,加入交流群,交个朋友吧,一起学习,一起进步!