加载模型生成文本
通过 Python 接口加载受支持的模型,输入提示词后获取文本生成结果。
BAIZE AI RESOURCE
Litgpt
LitGPT 是一款面向大语言模型使用者和开发者的工具,支持加载模型生成文本,也提供预训练、微调和部署流程。它将模型试用与定制所需的操作集中在同一项目中,适合需要自行运行或调整模型的用户。

LitGPT 是一款用于运行和定制大语言模型的工具,适合希望自行试用模型、用数据调整模型表现,或将模型提供给其他应用使用的用户。安装后,可以通过 Python 加载受支持的模型,输入提示词并获取生成结果;项目还提供预训练、微调、评估和部署相关流程。微调方面,仓库提供全量微调、LoRA 和 QLoRA 等配置,覆盖 Llama、Gemma、Phi 等模型系列。它不是打开网页即可对话的成品应用,使用者需要准备 Python 环境;训练或部署时,还需根据所选模型和任务准备相应的计算资源。
通过 Python 接口加载受支持的模型,输入提示词后获取文本生成结果。
提供全量微调、LoRA 和 QLoRA 相关流程及模型配置,便于按任务和资源条件选择调整方式。
项目提供预训练和模型评估流程,可用于训练模型并检验其表现。
提供模型服务部署流程,便于将运行中的模型接入其他应用。
使用场景开发者想测试模型能否修改句子中的拼写错误。
安装 LitGPT 后,通过 Python 加载 microsoft/phi-2,将待修改的句子作为提示词提交,查看模型生成的修改结果。
使用场景模型使用者想根据自己的任务数据调整 Gemma 模型。
从项目提供的 Gemma 微调配置中选择 LoRA 或 QLoRA 方案,再运行对应的微调流程,获得经过任务数据调整的模型。