大语言模型生态系统:助你自由调教 AI 模型

这些开源项目都是在语言模型领域具有重要影响力的优秀项目。它们共同的特点是强调了对大规模语言模型进行训练和推理的高效性、灵活性和可扩展性。无论是通过提供定制化的语言模型、支持并行计算和分布式训练,还是通过优化内存管理和硬件资源利用效率来提高运算速度,这些项目都致力于使得人工智能技术更加便捷、高效地应用于各个领域。如果您正在寻找一个功能强大且易于使用的开源语言模型项目,我强烈推荐阅读此篇文章。

nomic-ai/gpt4all

Stars: 50.0k License: MIT

gpt4all 是一个开源的边缘大型语言模型的生态系统。目标是成为任何个人或企业都可以自由使用,分发和构建的最佳指令调整助手式语言模型。

该项目具有以下核心优势:

  • 强大且定制化:gpt4all 提供了强大且可自定义的语言模型,可以在消费级 CPU 上本地运行。
  • 开放性:GPT4All 型号是一个 3GB - 8GB 的文件,您可以下载并插入到 GPT4All 开源生态系统软件中。Nomic AI支持和维护这个软件生态系统,以加强质量和安全性,同时带头努力允许任何人或企业轻松训练和部署自己的边缘大语言模型。
  • 桌面应用程序:通过桌面聊天客户端,在家庭台式电脑上原生运行任意 gpt4all 模型。您可以查看 gpt4all 网站以获取完整列表,并使用这款功能强大的桌面应用程序运行各种开源模型。

hpcaitech/ColossalAI

Stars: 31.4k License: Apache-2.0

Colossal-AI 是一个使得大型 AI 模型更便宜、更快速和更易于使用的开源项目。

主要功能:

  • 提供并行组件集合,支持用户以类似在本地机器上编写模型的方式进行分布式训练和推理。
  • 支持多种并行策略:数据并行、流水线并行、1D/2D/2.5D/3D 张量并行以及序列并行等。
  • 提供 Zero Redundancy Optimizer (ZeRO) 和自动化 Parallelism 等特性,提高效率和可扩展性。
  • 具备友好的使用体验,并基于配置文件实现了简单且灵活的代码编写风格。

关键特性与核心优势:

  • 并发能力强大:通过各种形式的张量级别或任务级别划分来充分利用硬件资源,在保证运算质量不降低情况下加速计算过程;
  • 内存管理智能高效:采用异构内存管理技术 PatrickStar 及其他方法有效减少 GPU 内存消耗;
  • 易于部署与应用广泛:ColossalAI 已经成功应用在聊天模型克隆、AIGC 模型加速以及生物医学领域的 AlphaFold 蛋白结构预测等多个实际场景中;
  • 提供丰富示例与文档:项目提供了大量使用案例和详细文档,方便用户快速上手并解决问题。

Colossal-AI 是一个非常有潜力的开源项目,它通过优化训练过程和硬件资源利用效率来使得 AI 模型更具成本效益。无论是在研究还是工业应用中,都能够帮助开发者们更高效地进行深度学习任务。

artidoro/qlora

Stars: 6.9k License: MIT

QLoRA 是一个高效的预训练语言模型 (LLM) 微调方法,可以在单个 48GB GPU 上减少内存使用量,并保持完整的 16-bit 微调任务性能。它通过将梯度反向传播到冻结的、4-bit 的量化预训练语言模型中来实现这一目标,并利用 Low Rank Adapters (LoRA) 进行后期微调。

以下是 QLoRA 的关键特性和核心优势:

  • 4-bit NormalFloat (NF4):引入了新的数据类型 NF4,对于正态分布权重而言,在信息论上达到最佳状态。
  • Double Quantization:通过对量化常数进行再次量化,降低平均内存占用。
  • Paged Optimizers:管理内存峰值以节省资源。

QLoRA 在超过 1000 个模型上进行了微调,并提供了详尽的指令遵循和聊天机器人表现分析结果。研究发现,在小规模高质量数据集上使用 QLoRa 进行微调可获得领先水平成果,即使使用了比之前的 SoTA 更小的模型也如此。

langgenius/dify

Stars: 6.2k License: NOASSERTION

Dify 是一个易用的 LLMOps 平台,旨在让更多人可以创建可持续运营的原生 AI 应用。Dify 提供多种类型应用的可视化编排,应用不仅开箱即用,也能以“后端即服务”的 API 提供服务。

  • 适配 Langchain,支持多个 LLMs
  • 开箱即用的的 Web 站点,支持表单模式和聊天对话模式
  • 一套 API 即可包含插件、上下文增强等能力,替你省下了后端代码的编写工作
  • 可视化的对应用进行数据分析,查阅日志或进行标注

未来计划:

  • Datasets:支持更多数据集 (例如从 Notion 或网页同步内容)

    我们将支持包括文本、网页甚至 Notion 内容在内的更多数据集。用户可以基于自己的数据源构建 AI 应⽤。

  • Plugins:引入符合 ChatGPT Plugin 标准或 Dify 自身生成 plugins。

    我们将发布符合 ChatGPT 标准或者由 Dify ⾃身产出 plugins 来实现扩展能力。

  • Open-source models,如采纳 Llama 作为模型提供商或进⼀步微调。

    我们会与像 Llama 这样优秀的开源模型合作,通过在我们平台中提供它们作为模型选项或者进⼀步微调来使之发挥更大价值。

Dify 是一个强大而简单的 LLMOps 工具。您可以使用 Dify 构建商业级应用程序和个人助手。如果你想要开发自己的应用程序,LangDifyGenius 可以帮助你节省后端工作并且提供可视化操作能力,从而让您不断改善和训练 GPT 模型。

bentoml/OpenLLM

Stars: 5.1k License: Apache-2.0

OpenLLM 是一个用于在生产环境中操作大型语言模型 (LLMs) 的开放平台。

主要功能:

  • 支持各种最先进的开源 LLMs,包括 StableLM、Falcon、Dolly 等。
  • 提供灵活的 API,可以通过 RESTful API 或 gRPC 进行服务,并支持 WebUI、CLI 和 Python/Javascript 客户端查询。
  • 可以自由构建 AI 应用程序,与 LangChain、BentoML 和 Hugging Face 无缝集成。
  • 简化部署流程,可自动生成 LLM 服务器 Docker 镜像或作为 Serverless 端点部署。

vllm-project/vllm

Stars: 4.4k License: Apache-2.0

vLLM 是一个快速且易于使用的库,用于 LLM 推理和服务。

以下是 vLLM 的核心优势:

  • 具有最先进的推理吞吐量
  • 使用 PagedAttention 高效管理注意力键值内存
  • 进行连续批处理来处理传入请求
  • 优化过的 CUDA 内核

vLLM 具备以下关键特性:

  • 与流行的 HuggingFace 模型无缝集成,灵活易用。
  • 具有各种解码算法的高通量服务,包括并行采样、波束搜索等。
  • 分布式推理的张量并行支持。
  • vLMM 完全兼容 OpenAI API Server 规范。

此外,vLLM 还可以无缝地支持许多 Huggingface 模型。例如 BLOOM、GPT2、GPT BigCode 等,并且在性能方面超越了 HuggingFace Transformers (HF) 和 Text Generation Inference (TGI),速度最高可以达到 HF 的 24 倍、TGI 的 3.5 倍。

mosaicml/llm-foundry

Stars: 2.9k License: Apache-2.0

LLM Foundry 是一个用于训练 MosaicML 基础模型的代码库。它旨在提供易于使用、高效和灵活的功能,以便快速尝试最新技术。

以下是该项目的核心优势和关键特性:

  • 提供了各种模型、数据集、回调函数和实用工具等源代码。
  • 包含运行 LLM 工作负载所需的脚本。
  • 可将原始文本数据转换为 StreamingDataset 格式,并存储在 data_prep/ 目录下。
  • 支持从 125M 到 70B 参数范围内对 HuggingFace 和 MPT 模型进行训练或微调。
  • 可通过 benchmarking 文件夹中提供的工具评估培训吞吐量、MFU (Most Frequently Used) 等指标。
  • 能够将模型转换为 HuggingFace 或 ONNX 格式,并生成响应结果,位于 inference/ 目录下。
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,547评论 6 477
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,399评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,428评论 0 337
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,599评论 1 274
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,612评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,577评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,941评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,603评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,852评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,605评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,693评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,375评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,955评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,936评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,172评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 43,970评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,414评论 2 342

推荐阅读更多精彩内容