xturingLLM 个性化微调工具

联合创作 · 2023-09-25 23:14

xturing 为 LLM 提供了快速、高效和简单的微调,如 LLaMA、GPT-J、GPT-2、OPT、Cerebras-GPT、Galactica 等。通过提供一个易于使用的界面,再根据你自己的数据和应用来个性化 LLM,xTuring 使构建和控制 LLM 变得简单。整个过程可以在你的电脑内或在你的私有云中完成,确保数据的隐私和安全。

通过 xturing,你可以:

  • 从不同的来源摄取数据,并将其预处理成 LLM 可以理解的格式
  • 从单个 GPU 扩展到多个 GPU,以便更快地进行微调
  • 利用内存效率高的技术(即 LoRA 微调)来减少你的硬件成本,最多可减少 90% 的时间。
  • 探索不同的微调方法,并以它们为基准,找到性能最好的模型
  • 在明确定义的指标上评估微调模型,进行深入分析

CLI

UI

安装

pip install xturing

快速开始

from xturing.datasets import InstructionDataset
from xturing.models import BaseModel

# Load the dataset
instruction_dataset = InstructionDataset("./alpaca_data")

# Initialize the model
model = BaseModel.create("llama_lora")

# Finetune the model
model.finetune(dataset=instruction_dataset)

# Perform inference
output = model.generate(texts=["Why LLM models are becoming so important?"])

print("Generated output by the model: {}".format(output))

 

 

浏览 10
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报