项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-10 · 当前星标 ★ 9429
技术标签
使用教程
ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。
环境要求
- Python 环境,可使用 uv 或 pip 安装 Python 包
- 若采用 Docker 镜像方式,需要 NVIDIA GPU 与容器运行时(命令中使用 --gpus all)
- 若在云端跑任务,需要 AWS、Azure、GCP 或 Lambda 等平台账号
- 只想快速试用的话,可用浏览器直接打开官方 Colab Notebook
安装与启动步骤
-
1安装 oumi
README 推荐使用 pip(uv pip install)方式安装 oumi 包,安装完成后即可使用 oumi 命令行。
uv pip install oumi -
2或用 Docker 镜像
不想在本地装依赖时,可拉取官方镜像并以交互方式运行 oumi 命令,注意该命令需要 GPU 环境。
docker run --gpus all -it ghcr.io/oumi-ai/oumi:latest oumi --help -
3验证命令可用
执行 oumi 的帮助命令,能正常输出 CLI 帮助说明即表示安装或镜像启动成功。
oumi --help -
4跑通入门示例
README 提供多个可直接在 Colab 打开的 Notebook,如 A Tour、微调教程、评估、蒸馏等,适合零环境快速体验。
-
5提交远程任务
需要更大算力时,可用 oumi launch 在 AWS、Azure、GCP、Lambda 等云平台运行任务,具体参数请参考官方文档。
oumi launch
如何确认成功
执行 oumi --help 能正常输出 CLI 帮助信息,或 docker run 后成功进入 oumi 命令行,即表示部署成功。
常见问题
Q:必须用 uv 安装吗?
A:README 的安装章节标题为“Using pip (Recommended)”,给出的命令是 uv pip install oumi,属于 pip 体系安装方式;如本机没有 uv,也可按同等 pip 语义安装。
Q:没有 GPU 能用吗?
A:Docker 方式给出的命令带 --gpus all,说明需要 GPU 环境;没有本地 GPU 时,可改用 README 提供的 Colab Notebook 试用。
Q:怎么快速体验微调和评估?
A:直接打开 README 中列出的 Colab Notebook,例如 Model Finetuning Guide(LoRA 微调)和 Model Evaluation,无需本地配置环境。
Q:怎么把任务放到云上跑?
A:使用 oumi launch 命令,README 说明其支持 AWS、Azure、GCP、Lambda 等平台,具体子命令与参数需查阅官方文档。
注意事项
- README 安装章节非常简短,仅给出 pip(uv pip install oumi)与 Docker 镜像两种入口,未列出完整依赖清单。
- Docker 镜像地址为 ghcr.io/oumi-ai/oumi:latest,运行示例中使用 --gpus all,请确认本机已安装 NVIDIA 驱动与容器 GPU 支持。
- 实际训练、评估、部署的详细配置项 README 节选中未展开,建议结合官网 oumi.ai 与仓库中的笔记本、示例配置使用。
核心亮点
- 统一接口覆盖微调、评估、部署全流程,减少工具切换成本
- 支持多种主流开放权重模型(Qwen、Llama、Gemma、GPT-OSS)
- 内置 DPO 等对齐算法,方便快速进行偏好优化
不足之处
- 文档/社区待观察
- 高级功能(如自定义训练器)可能需要深入阅读源码
适用场景
- 针对特定领域数据对开源 LLM 进行快速微调
- 在多个基准上对比不同模型的性能以选择最优
- 将微调后的模型部署为内部 API 服务供业务调用
替代项目
LLaMA-Factory、Axolotl、Hugging Face TRL
下一篇:Kiln
同类项目推荐
determined
开源
一键拉起分布式训练,自动调参省心省力
Determined is an open-source machine learning platform that simplifies distributed t···
MPO
开源
让LLM智能体先学会定计划,再动手执行,提升复杂任务成功率
MPO: Boosting LLM Agents with Meta Plan Optimization (EMNLP 2025 Findings)
MLM_Filter
开源
用多模态模型自动筛出高质量图文对,提升训练数据纯净度
Official implementation of our paper "Finetuned Multimodal Language Models are High-···
MoE-plus-plus
开源
零计算专家,让MoE模型更快更省,性能不减
[ICLR 2025] MoE++: Accelerating Mixture-of-Experts Methods with Zero-Computation Exp···
mini-verl
开源
单张消费级显卡就能跑 PPO/GRPO 对齐训练
verl for a single consumer GPU. PPO, GRPO and on-policy distillation on NVIDIA GPUs.
ParScale
开源
并行跑多个模型,不堆参数也能提升性能
Parallel Scaling Law for Language Model — Beyond Parameter and Inference Time Scali···
CICD-for-Machine-Learning
开源
用GitHub Actions一键跑通模型训练到部署全流程
A beginner's project on automating the training, evaluation, versioning, and deploym···
chatgpt-finetune-ui
开源
拖拽上传数据,点点鼠标就能微调 ChatGPT 模型
Simple python WebUI for fine-tuning ChatGPT (gpt-3.5-turbo)
LoongSage
开源
用验证过的配方,快速训出会调工具的智能体
A production-grade, high-efficiency Agentic RL framework for frontier LLMs with vali···
