LongLoRA

用少量算力把大模型上下文拉到 100k+,长文本任务轻松搞定

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 开源模型 🍴 复刻 278 🐛 待处理 53 👀 关注 12
GitHub ↗
GitHub 星标 ★ 2685
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,Apache-2.0 许可证,代码和文档公开,可免费使用和部署。

项目预览

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

项目数据

分类开源模型
开发团队JIA-Lab-research
定价模式🏷️完全免费
价格说明开源项目,Apache-2.0 许可证,代码和文档公开,可免费使用和部署。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议Apache-2.0
主要语言Python
技术栈/模型fine-tuning-llm,large-language-models,llm,long-context,lora
GitHub 星标★ 2685
上线时间2023-09-21 00:00:00
代码更新2024-08-14
仓库体积11.4 MB
本站数据更新2026-11-07
维护状态低维护
收录时间2026-08-09
浏览次数6

项目介绍

LongLoRA 是一个针对大语言模型的长上下文微调方案,由 MIT 等机构提出,相关论文入选 ICLR 2024 Oral。它解决了在有限算力下将预训练模型(如 Llama 2)的上下文长度从 4k 扩展到 100k+ 的难题。核心创新包括:提出可转移的稀疏注意力机制(Shifted Sparse Attention,S2Attn),在微调阶段近似全注意力,大幅降低计算开销;同时结合 LoRA 低秩适配,仅更新少量参数即可实现高效长上下文扩展。项目提供了完整的代码、训练脚本和预训练权重,支持 Llama 2 和 Llama 3 等模型,并附带 LongAlpaca 长指令微调数据集。用户可基于此快速将现有模型扩展为长文本模型,用于长文档理解、多轮对话等场景。

Star 增长趋势

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

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

使用教程

ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。

难度:进阶 约 30 分钟 部署方式:本地安装 6 步

环境要求

  • Python 环境与 pip(用于安装 requirements.txt 依赖)
  • Hugging Face 账号,且注册邮箱有效
  • 已接受 Meta 的 Llama 许可与可接受使用政策(下载预训练权重前必须完成)
  • 具备可用于微调/推理的 GPU 环境(README 未给出具体显存要求)

安装与启动步骤

  1. 1Fork 项目仓库

    登录 GitHub,打开项目主页点击 Fork,把仓库复制到自己的账号下,便于后续修改和保存训练脚本。

  2. 2克隆仓库到本地

    在自己的机器上执行 git clone,把 Fork 后的仓库拉到本地;如已在本地有代码可跳过此步。

    git clone https://github.com/JIA-Lab-research/LongLoRA.git
  3. 3安装 Python 依赖

    进入项目目录后安装 requirements.txt 中的依赖,建议先激活虚拟环境避免污染系统 Python。

    pip install -r requirements.txt
  4. 4安装 flash-attn

    按要求使用 --no-build-isolation 参数安装 flash-attn,否则可能因构建隔离导致安装失败,编译耗时较长需耐心等待。

    pip install flash-attn --no-build-isolation
  5. 5准备模型权重

    二选一:直接使用官方 Released model 权重,或按 README 的 Fine tune 流程自行微调一个适合自己偏好的模型。

  6. 6对话测试与部署

    通过 chat 方式测试模型效果,确认长上下文表现正常后,再参考 Demo 章节部署自己的演示服务。

如何确认成功

按 README 建议用 chat 方式与模型对话,能正常加载权重并正确回应即视为部署成功。

常见问题

Q:下载预训练权重前需要做什么?

A:需要一个有效邮箱注册的 Hugging Face 账号,并用同一邮箱接受 Meta 的 Llama 许可与可接受使用政策,否则无法下载权重。

Q:flash-attn 安装失败怎么办?

A:务必按 README 使用 --no-build-isolation 参数安装;该包需要本地编译,确认已装好匹配的 CUDA 与编译工具链。

Q:一定要自己训练模型吗?

A:不必。README 说明可以直接使用官方已发布(Released model)的权重,也可以按 Fine tune 流程自行微调。

Q:怎么确认长上下文能力?

A:README 提到 LongAlpaca 系列模型在 LongBench 和 L-Eval 基准上有评测结果,可参考官方 benchmarks 目录进行对比验证。

注意事项

  • 仓库同时提供训练脚本、预训练权重与 LongAlpaca 长指令数据集,按需选用。
  • 训练和推理对算力要求较高,建议先确认本地 GPU 资源再开始。
  • README 未给出具体依赖版本、端口与配置项,实际参数请以仓库内文件为准。

核心亮点

  • S2Attn 稀疏注意力机制在微调时高效模拟全注意力,训练成本显著低于全参微调
  • 与 LoRA 结合,仅需微调少量参数即可扩展上下文,普通 GPU 也能跑
  • 提供完整代码、预训练权重和 LongAlpaca 数据集,开箱即用

不足之处

  • 推理时仍使用标准注意力,长上下文下的显存占用和延迟问题未根本解决
  • 主要针对 Llama 系列模型,对其他架构的适配性有限

适用场景

  • 长文档摘要与问答(如论文、合同)
  • 多轮长对话系统(如客服、助手)
  • 代码库级上下文理解(如跨文件代码分析)

替代项目

YaRN、Position Interpolation、FlashAttention

上一篇:Medusa

下一篇:xTuring

同类项目推荐

FreeFuse 开源

测试时动态路由,多 LoRA 融合不打架

[TMLR26]FreeFuse: Multi-Subject LoRA Fusion via Adaptive Token-Level Routing at Test···

★ 249 2026-08-09
EasyWAM 开源

一套代码搞定世界动作模型的训练、微调与评测

A unified framework for training, fine-tuning, and evaluating World Action Models

★ 405 2026-09-16
mlops-hf-tf-vision-models 开源

把Hugging Face视觉模型一键部署到生产流水线

MLOps for Vision Models (TensorFlow) from Transformers with TensorFlow Extended (TF···

★ 111 2026-08-09
DrMAS 开源

把多智能体协作变成可训练环境,让多个 LLM 一起进化

Dr. MAS is an end-to-end RL training framework for multi-agent LLM systems, supporti···

★ 172 2026-10-07
WaDi 开源

一步生成高质量图像,蒸馏时精准把控权重方向

[CVPR 2026] Official PyTorch implementation of WaDi: Weight Direction-aware Distilla···

★ 128 2026-08-09
vime 开源

用 vLLM 加速 LLM 强化学习后训练,采样更快、规模更大

An LLM post-training framework with vLLM for RL Scaling

★ 503 2026-10-07
Unified-MoE-Compression 开源

给 MoE 模型瘦身,压缩后推理更快更省内存

The official implementation of the paper "Towards Efficient Mixture of Experts: A Ho···

★ 93 2026-08-09
CUDA-L2 开源

用强化学习自动写出比 cuBLAS 更快的矩阵乘法内核

CUDA-L2: Surpassing cuBLAS Performance for Matrix Multiplication through Reinforceme···

★ 487 2026-08-09
MLOps 开源

跟着项目清单,动手练熟MLOps全流程

The repository contains a list of projects which I will work on while learning and i···

★ 79 2026-08-09