一句话:通过在 vLLM 与 HuggingFace pipelines 上评测 20 个开源大模型的吞吐,揭示超参数配置对推理性能的关键作用。
开源大语言模型让开发者能在掌控隐私与合规的前提下构建 AI 应用,但需要推理引擎将模型权重加载到 GPU 等资源并处理查询,而推理速度对实时应用至关重要。本文以吞吐量(单位时间生成的 token 数)为核心指标,对 20 个开源大语言模型在 vLLM 与 HuggingFace pipelines 两种推理库上的性能进行系统评测,考察开发者必须配置的多种超参数如何影响推理性能。结果表明,超参数优化对达到最佳性能不可或缺;在 GPU 升级或降级时重新进行超参数调优也能提升吞吐,例如在 HuggingFace pipelines 中平均分别提升 9.16% 与 13.7%。
Open-source large language models (LLMs) enable developers to create AI-based solutions while maintaining control over aspects such as privacy and compliance, thereby providing governance and ownership of the model deployment process. To utilize these LLMs, inference engines are needed. These engines load the model's weights onto available resources, such as GPUs, and process queries to generate responses. The speed of inference, or performance, of the LLM is critical for real-time applications. In this paper, we analyze the performance, particularly throughput (tokens generated per unit of time), of 20 LLMs using two inference libraries: vLLM and HuggingFace's pipelines. We investigate how various hyper-parameters, which developers must configure, influence inference performance. Our results reveal the importance of hyperparameter optimization to achieve maximum performance. We demonstrate that applying hyperparameter optimization during GPU upgrades or downgrades can enhance throughput. For instance, in Hugging-Face pipelines, this approach yields average improvements of 9.16% and 13.7%, respectively.