onebitllms

小显存也能训练1.58bit三值大模型

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 模型训练 🍴 复刻 14 🐛 待处理 1 👀 关注 1
GitHub ↗
GitHub 星标 ★ 150
维护状态 维护中
是否开源 开源
定价模式 🏷️完全免费 开源免费工具包,无在线服务

项目预览

onebitllms 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类模型训练
开发团队tiiuae
定价模式🏷️完全免费
价格说明开源免费工具包,无在线服务
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议NOASSERTION
主要语言Python
GitHub 星标★ 150
上线时间2025-04-16 00:00:00
代码更新2026-08-18
仓库体积103 KB
本站数据更新2026-11-03
维护状态维护中
收录时间2026-09-16
浏览次数1

项目介绍

onebitllms 是一个轻量级 Python 工具包,专注于训练和微调 1.58bit 量化语言模型。传统大模型动辄需要数十 GB 显存,普通开发者难以在消费级显卡上训练或微调。该项目通过极低比特量化(约 1.58bit,即三值权重 -1/0/1)大幅压缩模型体积与计算开销,让单卡甚至小显存设备也能跑通训练与微调流程。核心能力包括:提供简洁的训练与微调接口,封装量化模型的前向与反向逻辑,降低从零搭建三值网络的门槛。它适合作为研究极低比特训练可行性的实验底座,也方便个人开发者在有限算力下验证想法。项目处于早期阶段,代码量小、依赖轻,便于阅读和二次改造,但生态与文档尚在完善中。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-09-16 · 当前星标 ★ 150

核心亮点

  • 专注1.58bit三值量化训练,显存占用远低于常规FP16训练
  • Python轻量工具包,接口简洁,易于阅读和二次开发
  • 覆盖训练与微调两条链路,适合做极低比特实验的起点

不足之处

  • 项目处于早期,文档和示例较少,上手需读源码
  • 社区规模小,缺少预训练权重和成熟生态支持

适用场景

  • 个人开发者用单张消费级显卡实验三值模型训练
  • 研究人员验证1.58bit量化对模型效果的影响
  • 小团队在有限算力下微调轻量语言模型做原型验证

替代项目

bitsandbytes、GPTQ-for-LLaMa

同类项目推荐

FineEnvs 开源

快速搭 LLM 强化学习环境,从定义到训练一条龙

FineEnvs — RL Environments 101: building and scaling RL environments in the age of ···

★ 545 2026-09-27
CorDA 开源

用上下文感知的低秩分解,让大模型微调更省资源更精准

CorDA: Context-Oriented Decomposition Adaptation of Large Language Models for task-a···

★ 60 2026-08-09
INTERS 开源

给大模型注入搜索能力,让检索结果更精准

This is the repository for our paper "INTERS: Unlocking the Power of Large Language ···

★ 208 2026-08-09
MegaDLMs 开源

用扩散模型训练语言模型,更快更灵活

GPU-optimized framework for training diffusion language models at any scale. The bac···

★ 343 2026-08-09
diffusers-torchao 开源

一键优化扩散模型,推理提速3倍,显存减半

End-to-end recipes for optimizing diffusion models with torchao and diffusers (infer···

★ 396 2026-08-09
labs-molt 开源

用HuggingFace生态跑智能体强化学习,省心扩展

A scalable, agentic-first, and HuggingFace-native RL framework for research (9k line···

★ 1210 2026-10-04
ms-swift 开源

一条命令微调千种大模型,从训练到部署全搞定

Use PEFT or Full-parameter to CPT/SFT/DPO/GRPO 600+ LLMs (Qwen3.8, DeepSeek-V4, GLM-···

★ 16020 2026-08-09
traceml 开源

一键诊断 PyTorch 训练瓶颈,提升 GPU 利用率

Find why PyTorch training is slow, compare runs, and catch performance regressions i···

★ 189 2026-08-09
SuperPiano 开源

在 Colab 里用 Transformer 一键生成钢琴曲

Absolutely amazing SOTA Google Colab (Jupyter) Notebooks for creating/training SOTA ···

★ 91 2026-08-09