离线批处理骨架
离线批处理骨架
from vllm import LLM, SamplingParams
llm = LLM(model="<模型路径>")
sp = SamplingParams(temperature=0.2, max_tokens=512)
prompts = ["任务一……", "任务二……"]
outs = llm.generate(prompts, sp)
for o in outs:
print(o.outputs[0].text[:80])离线与在线分开部署:离线任务把批参数调大换吞吐,避免与在线服务抢 KV 与算力。