BAIZE AI RESOURCE

大语言模型训练与推理工具

Litgpt

LitGPT 是一款面向大语言模型使用者和开发者的工具,支持加载模型生成文本,也提供预训练、微调和部署流程。它将模型试用与定制所需的操作集中在同一项目中,适合需要自行运行或调整模型的用户。

Litgpt

资源介绍

LitGPT 是一款用于运行和定制大语言模型的工具,适合希望自行试用模型、用数据调整模型表现,或将模型提供给其他应用使用的用户。安装后,可以通过 Python 加载受支持的模型,输入提示词并获取生成结果;项目还提供预训练、微调、评估和部署相关流程。微调方面,仓库提供全量微调、LoRA 和 QLoRA 等配置,覆盖 Llama、Gemma、Phi 等模型系列。它不是打开网页即可对话的成品应用,使用者需要准备 Python 环境;训练或部署时,还需根据所选模型和任务准备相应的计算资源。

应用能力

加载模型生成文本

通过 Python 接口加载受支持的模型,输入提示词后获取文本生成结果。

模型微调

提供全量微调、LoRA 和 QLoRA 相关流程及模型配置,便于按任务和资源条件选择调整方式。

预训练与评估

项目提供预训练和模型评估流程,可用于训练模型并检验其表现。

模型部署

提供模型服务部署流程,便于将运行中的模型接入其他应用。

资源配置

应用版本

v0.5.13

运行环境

Linux

安装方式

解压即用

依赖环境

Python

安装与使用

  1. 准备 Python 3.10 或更高版本的环境,并根据所选模型准备计算资源。

  2. 运行 pip install 'litgpt[extra]' 安装 LitGPT。

  3. 先通过 Python 接口加载受支持的模型,输入提示词,获取生成文本。

  4. 如需定制或提供模型服务,再选择相应的微调、评估或部署流程。

使用案例

试用文本生成

使用场景开发者想测试模型能否修改句子中的拼写错误。

安装 LitGPT 后,通过 Python 加载 microsoft/phi-2,将待修改的句子作为提示词提交,查看模型生成的修改结果。

选择模型微调配置

使用场景模型使用者想根据自己的任务数据调整 Gemma 模型。

从项目提供的 Gemma 微调配置中选择 LoRA 或 QLoRA 方案,再运行对应的微调流程,获得经过任务数据调整的模型。

包含内容

release-0-litgpt-0.5.13.tar.gz
README-中文安装说明.md
LICENSE-Apache-2.0.txt
SHA256.txt

常见问题

LitGPT 需要什么 Python 版本?
项目配置要求 Python 3.10 或更高版本。
可以用 LitGPT 直接生成文本吗?
可以。README 给出的用法是通过 Python 的 LLM.load 加载模型,再调用 generate 传入提示词;示例使用 microsoft/phi-2。
LitGPT 提供哪些微调方式?
仓库提供全量微调、LoRA 和 QLoRA 的相关流程与配置;可用配置因模型而异。