部署例程
人工智能算法部署
AIBOX-1684 具备全面的人工智能算法本地私有化部署能力,无论是新颖、高算力需求的大语言模型,还是经典的 YOLOv5 目标检测模型,AIBOX-1684 均可胜任。
本章节主要介绍 SOPHON-DEMO 项目,大模型部署请参考大模型部署这一章节。
SOPHON-DEMO 项目简介
SOPHON-DEMO 包含了一系列主流人工智能算法的移植例程,以及详细的部署文档以便于用户运行。
项目仓库链接:SOPHON-DEMO。
例程清单
SOPHON-DEMO 提供的例子从易到难分为 tutorial、sample、application 三个模块:
tutorial模块存放一些基础接口的使用示例。sample模块存放一些经典算法在SOPHONSDK上的串行示例。application模块存放一些典型场景的典型应用。
| tutorial | 编程语言 |
|---|---|
| resize | C++/Python |
| crop | C++/Python |
| crop_and_resize_padding | C++/Python |
| ocv_jpubasic | C++/Python |
| ocv_vidbasic | C++/Python |
| blend | C++/Python |
| stitch | C++/Python |
| sample | 算法类别 | 编程语言 | BModel |
|---|---|---|---|
| LPRNet | 车牌识别 | C++/Python | FP32/FP16/INT8 |
| ResNet | 图像分类 | C++/Python | FP32/FP16/INT8 |
| RetinaFace | 人脸检测 | C++/Python | FP32 |
| segformer | 语义分割 | C++/Python | FP32/FP16 |
| SAM | 语义分割 | Python | FP32/FP16 |
| yolact | 实例分割 | C++/Python | FP32/FP16/INT8 |
| YOLOv8_seg | 实例分割 | C++/Python | FP32/FP16/INT8 |
| PP-OCR | OCR | C++/Python | FP32/FP16 |
| OpenPose | 人体关键点检测 | C++/Python | FP32/FP16/INT8 |
| C3D | 视频动作识别 | C++/Python | FP32/FP16/INT8 |
| DeepSORT | 多目标跟踪 | C++/Python | FP32/FP16/INT8 |
| ByteTrack | 多目标跟踪 | C++/Python | FP32/FP16/INT8 |
| CenterNet | 目标检测、姿态识别 | C++/Python | FP32/FP16/INT8 |
| YOLOv5 | 目标检测 | C++/Python | FP32/FP16/INT8 |
| YOLOv34 | 目标检测 | C++/Python | FP32/INT8 |
| YOLOX | 目标检测 | C++/Python | FP32/INT8 |
| SSD | 目标检测 | C++/Python | FP32/INT8 |
| YOLOv7 | 目标检测 | C++/Python | FP32/FP16/INT8 |
| YOLOv8_det | 目标检测 | C++/Python | FP32/FP16/INT8 |
| YOLOv5_opt | 目标检测 | C++/Python | FP32/FP16/INT8 |
| ppYOLOv3 | 目标检测 | C++/Python | FP32/FP16/INT8 |
| ppYoloe | 目标检测 | C++/Python | FP32/FP16 |
| WeNet | 语音识别 | C++/Python | FP32/FP16 |
| BERT | 语言模型 | C++/Python | FP32/FP16 |
| ChatGLM2 | 语言模型 | C++/Python | FP16/INT8/INT4 |
| Llama2 | 语言模型 | C++/Python | FP16/INT8/INT4 |
| ChatGLM3 | 语言模型 | Python | FP16/INT8/INT4 |
| Qwen | 语言模型 | Python | FP16/INT8/INT4 |
| Qwen1_5 | 语言模型 | Python | FP16/INT8/INT4 |
| StableDiffusionV1.5 | 图像生成 | Python | FP32/FP16 |
| GroundingDINO | 多模态目标检测 | Python | FP16 |
| application | 应用场景 | 编程语言 |
|---|---|---|
| VLPR | 多路车牌检测+识别 | C++/Python |
| YOLOv5_multi | 多路目标检测 | C++ |
| YOLOv5_multi_QT | 多路目标检测+QT_HDMI显示 | C++ |
大模型部署
LLM-TPU 项目简介
LLM-TPU 包含了各类开源生成式 AI 模型的移植部署例程,其中以 LLM 为主,也包含 Stable Diffusion(AI 绘画)。
项目仓库链接:LLM-TPU。
例程清单
| Model | INT4 | INT8 | FP16/BF16 | Huggingface Link |
|---|---|---|---|---|
| Baichuan2-7B | ✔ | LINK | ||
| ChatGLM3-6B | ✔ | ✔ | ✔ | LINK |
| CodeFuse-7B | ✔ | ✔ | LINK | |
| DeepSeek-6.7B | ✔ | ✔ | LINK | |
| Falcon-40B | ✔ | ✔ | LINK | |
| Phi-3-mini-4k | ✔ | ✔ | ✔ | LINK |
| Qwen-7B | ✔ | ✔ | ✔ | LINK |
| Qwen-14B | ✔ | ✔ | ✔ | LINK |
| Qwen-72B | ✔ | LINK | ||
| Qwen1.5-0.5B | ✔ | ✔ | ✔ | LINK |
| Qwen1.5-1.8B | ✔ | ✔ | ✔ | LINK |
| Llama2-7B | ✔ | ✔ | ✔ | LINK |
| Llama2-13B | ✔ | ✔ | ✔ | LINK |
| LWM-Text-Chat | ✔ | ✔ | ✔ | LINK |
| Mistral-7B-Instruct | ✔ | ✔ | LINK | |
| Stable Diffusion | ✔ | LINK | ||
| Stable Diffusion XL | ✔ | LINK | ||
| WizardCoder-15B | ✔ | LINK | ||
| Yi-6B-chat | ✔ | ✔ | LINK | |
| Yi-34B-chat | ✔ | ✔ | LINK |
RAG 私有知识库
检索增强生成(RAG)是一种使用来自私有或专有数据源的信息来辅助文本生成的技术,该技术通过结合大型语言模型(LLM)的生成能力与从外部知识库中检索相关信息的能力,来生成更加精准和上下文相关的回答或文本内容。
通过借助 RAG 技术,可以解决大模型存在的知识时效性不足、上下文理解限制、信息来源不确定等关键性问题。
本章节提供三个可部署于 AIBOX-1684 上的 RAG 部署案例,分别为 FireflyChat、ChatDoc-TPU 和 LangChain-Chatchat-TPU。
FireflyChat 项目简介
FireflyChat 是由 Firefly 开源团队开发的图形化大模型应用平台,部署仅需简单安装、无需编译,快速体验 RAG 对大模型的提升。
项目详情参考 FireflyChat 章节。
ChatDoc-TPU 项目简介
ChatDoc-TPU 是一个完全本地化推理的文档对话工具,其主要目标是通过使用自然语言来简化与文档的交互,并提取有价值的信息。
项目仓库链接:ChatDoc-TPU
LangChain-Chatchat-TPU 项目简介
Langchain-Chatchat-TPU 是基于 Langchain-Chatchat 开发的完全本地化推理的知识库增强方案。
项目仓库链接:LangChain-Chatchat-TPU
语音识别与文本转语音
VITS-TPU 项目简介
VITS 是一个端到端的文本转语音(TTS)模型,VITS-TPU 项目实现了对 VITS 的 TPU 算法移植。
项目仓库链接:VITS-TPU
Whisper-UI-TPU 项目简介
Whisper 是一个由 OpenAI 开发的开源语音识别(ASR)模型,它能够实现实时、多语言的语音识别。
Whisper-UI-TPU 实现了对 Whisper 的 TPU 算法移植,同时提供了便于使用的 WebUI。
项目仓库链接:Whisper-UI-TPU
图片内容检索
CLIP-TPU 项目简介
CLIP 全称 Constrastive Language-Image Pre-training,是 OpenAI 推出的采用对比学习的文本-图像预训练模型。CLIP 在 zero-shot 文本-图像检索,zero-shot 图像分类,文生图任务 guidance,open-domain 检测分割等任务上均有非常惊艳的表现。
CLIP-TPU 项目实现了对 CLIP 的 TPU 算法移植,用户可使用该项目验证 CLIP 在 AIBOX-1684 上的实际表现。
项目仓库链接:CLIP-TPU
ImageSearch-tpu 项目简介
ImageSearch-tpu 项目基于 CLIP 实现了对大量图片进行内容搜索的功能,同时提供了可视化的 WebUI 便于用户使用。
项目仓库链接:ImageSearch-tpu

