360 Secure SkillHub
360 安全技能中心

LoRA Toolkit

这是面向LLM的LoRA微调工具包,支持Llama、Mistral等HuggingFace兼容模型。核心功能包括:根据模型、GPU显存、数据集规模生成训练配置;估算显存占用、训练时间及成本;生成基于PEFT+TRL的可运行训练脚本;验证多种格式数据集;推荐适配模型。用户通过bash命令传参使用,适用于LLM微调前的准备工作。

loutai0307-prog
loutai0307-prog
热度 45.23
最近更新 2026-05-11 17:40:29
分类:开发依赖:纯提示词应用场景:代码生成 / 数据处理

安装技能

复制下面的话术发给你的 Agent,它会通过 secure-skills 检索并安装该技能。

执行命令安装此 skill: npx secure-skills-cli add bytesagain-lora-toolkit
Metadata
name
bytesagain-lora-toolkit
description
Configure, estimate, and generate LoRA fine-tuning scripts for LLMs. Input: base model name, dataset size, GPU spec. Output: training config, PEFT script, cost estimate.
version
1.0.0
author
BytesAgain
tags
["lora","fine-tuning","llm","machine-learning","training","peft","huggingface"]

LoRA Toolkit

Configure and generate LoRA fine-tuning scripts for large language models. Supports Llama, Mistral, Qwen, Phi and other HuggingFace-compatible models.

Commands

config

Generate a LoRA training configuration for your model and hardware.

bash scripts/script.sh config --model llama3-8b --gpu 24gb --dataset 10000

Parameters:

  • --model — base model (llama3-8b, mistral-7b, qwen2-7b, phi3-mini, llama3-70b)
  • --gpu — VRAM size (8gb, 16gb, 24gb, 40gb, 80gb)
  • --dataset — number of training samples

estimate

Estimate VRAM usage, training time, and cost before starting.

bash scripts/script.sh estimate --model mistral-7b --gpu 16gb --dataset 5000 --epochs 3

generate

Generate a ready-to-run Python training script using HuggingFace PEFT + TRL.

bash scripts/script.sh generate --model llama3-8b --output train.py

validate

Check dataset format compatibility (Alpaca / ShareGPT / OpenAI Chat format).

bash scripts/script.sh validate --file dataset.json --format alpaca

recommend

Recommend the best base model for your use case and hardware.

bash scripts/script.sh recommend --task chat --gpu 16gb --language en

help

Show all commands.

bash scripts/script.sh help

LoRA Parameters Reference

Model Size Recommended Rank Alpha VRAM (4-bit)
7B 16–32 32–64 8–12 GB
13B 16 32 14–18 GB
70B 8–16 16–32 40–48 GB

Supported Dataset Formats

  • Alpaca: {"instruction": "...", "input": "...", "output": "..."}
  • ShareGPT: {"conversations": [{"from": "human", "value": "..."}, ...]}
  • OpenAI Chat: {"messages": [{"role": "user", "content": "..."}, ...]}

Requirements

  • Python 3.8+
  • Optional: pip install transformers peft trl datasets for script execution

Feedback

https://bytesagain.com/feedback/ Powered by BytesAgain | bytesagain.com