microsoft/onnxruntime-genai

Generative AI with ONNX Runtime

解决的问题

提供一种高性能且灵活的方式,在本地设备上运行生成式 AI 模型(特别是 LLM),处理生成式 AI 的复杂循环,包括预处理、推理和后处理。

工作原理

实现了 ONNX 模型的生成式 AI 循环。包括管理 KV 缓存、logits 处理、搜索与采样,以及预处理/后处理。支持广泛的硬件加速选项(CPU、CUDA、DirectML、OpenVINO、QNN、WebGPU),并兼容多种模型架构,如 Llama、Phi、Mistral 和 Gemma。

适用人群

希望使用 ONNX Runtime 将生成式 AI 功能集成到应用程序中的开发者,目标操作系统包括 Linux、Windows、Mac、Android,硬件架构包括 x86、x64、arm64。

主要亮点

  • 支持多 LoRA
  • 支持约束和连续解码
  • 广泛支持模型架构,包括 Llama、Phi 和 Mistral
  • 支持 Python、C#、C/C++ 和 Java 的跨平台 API
  • 提供丰富的硬件加速选项,包括 CUDA 和 DirectML

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目