提交智能体收藏0
技术动态

AirLLM v4.0.0:以流式 LoRA 在 6GB 显存下训练 Qwen3.8-Flash-Next

AirLLM v4.0.0 引入 Streamed LoRA:模型冻结权重保留在磁盘,适配器驻留 GPU。发布说明称 Qwen3.8-Flash-Next 可在 6GB 显存的 RTX 3060 Ti 上训练,Qwen3.8-27B 在序列长度 512 时约需 2GB 显存。

版本变化

AirLLM v4.0.0 的发布说明介绍了 Streamed LoRA 的训练方式:模型原有的冻结权重继续留在磁盘上,只把 LoRA 适配器放到 GPU 显存中,从而降低训练所需的显存占用。

发布说明给出的显存条件

  • Qwen3.8-Flash-Next:可在 6GB 显存的 RTX 3060 Ti 上训练。
  • Qwen3.8-27B:在序列长度 512 时显存占用约为 2GB。

以上数据均为发布说明中的表述,具体表现取决于所用模型、数据与配置。

安装与训练方式

发布说明列出的前置操作为:

  • pip install -U airllm
  • 从 Hugging Face 的 transformers 仓库安装开发版本

示例脚本 air_llm/examples/train_qwen38_flash_next_lora.py 通过 --data 指向 JSONL 数据文件,支持两种记录格式:{"text": "..."} 或 {"prompt": "...", "completion": "..."}。脚本还提供 --seq-len、--epochs 和 --save-adapter 参数,分别用于设置序列长度、训练轮数和适配器保存路径。

已知限制

发布说明指出,Qwen3.8-Flash-Next 目前仍依赖 transformers 代码树内的 qwen4_exp,即需要 GitHub 上 transformers 主分支的版本。

来源与核对说明

  1. lyogavin/airllm ↗https://github.com/lyogavin/airllm/releases/tag/v4.0.0

本文经 AI 整理及独立自动校对,未做运行验证;自动校对仍可能遗漏问题,请以原始资料为准。

这篇内容对你有帮助吗?

点赞人数由服务器记录,收藏保存在当前浏览器。

继续阅读

Headroom v0.38.0 发布:beacon schema v2、路由省钱统计与多项压缩修复 →

Mem0 Pi Agent 插件 v0.3.2:调整记忆检索策略与自动召回提示 →

claude-mem v13.39.0:Codex 子代理线程压缩后恢复记忆时间线 →

Headroom v0.39.0 发布:压缩可解释性、逐消息保护与缓存修复 →