SuperAPI AI 工具集

LLaMA-O1

LLaMA-O1 是一个基于 PyTorch 和 HuggingFace 构建的开源大型推理模型框架,旨在支持训练、推理和评估。项目由 SimpleBerry 开发,并已发布多个预训练和微调模型及数据集,包括 LLaMA-O1-Supervised-1129 等。该框架致力于推动开源大型推理模型的发展,支持长链思维(Long CoT)标记、预训练、监督微调、过程奖励模型(PRM)令牌修正以及自玩强化学习等功能。部分功能如强化学习仍在训练中,推理时推理增强框架暂时推迟。用户可通过 HuggingFace 获取模型、数据集和在线演示(仅 CPU)。

访问官网

工具介绍

LLaMA-O1 是一个开源的大型推理模型框架,专注于训练、推理和评估,基于 PyTorch 和 HuggingFace 构建。项目由 SimpleBerry 维护,目标是为开源社区提供大型推理模型(Large Reasoning Models)的完整解决方案。

主要能力包括:

  • 长链思维(Long CoT)标记:已完成。
  • 预训练数据集:已完成,可在 HuggingFace 上获取(OpenLongCoT-Pretrain-1202)。
  • 监督微调数据集:已完成,可在 HuggingFace 上获取(OpenLongCoT-SFT)。
  • PRM 令牌修正数据集:已完成。
  • 自玩强化学习(Reinforcement Learning With Self-Play):代码已完成,正在训练中。
  • 推理时推理增强框架:代码已完成,暂时推迟。

使用方式: 用户可以通过 HuggingFace 获取模型和数据集。具体包括:

  • 基础模型(预训练):https://huggingface.co/SimpleBerry/LLaMA-O1-Base-1127
  • 监督微调模型:https://huggingface.co/SimpleBerry/LLaMA-O1-Supervised-1129
  • 监督微调数据集:https://huggingface.co/datasets/SimpleBerry/OpenLongCoT-SFT
  • 预训练数据集:https://huggingface.co/datasets/SimpleBerry/OpenLongCoT-Pretrain-1202
  • GGUF 格式模型:https://huggingface.co/Lyte/LLaMA-O1-Supervised-1129-Q4_K_M-GGUF
  • 在线演示(仅 CPU):https://huggingface.co/spaces/SimpleBerry/LLaMA-O1-Supervised-1129-Demo

此外,项目还提供了相关研究论文链接:

  • https://huggingface.co/papers/2406.07394
  • https://huggingface.co/papers/2410.02884
  • https://huggingface.co/papers/2411.18203

适用场景: LLaMA-O1 适用于研究人员和开发者进行大型推理模型的训练、微调、推理和评估。它特别适合需要长链思维推理能力的任务,如复杂问题求解、数学推理等。

限制:

  • 部分功能(如强化学习)仍在进行中,尚未完全发布。
  • 推理时推理增强框架暂时推迟。
  • 在线演示仅支持 CPU 运行,可能性能有限。
  • 具体训练细节、硬件要求、使用步骤等资料中暂未提供。