WhiteLightning 提供了一站式解决方案,只需单行命令即可完成。具体步骤如下:
- 安装 Docker 并拉取 WhiteLightning 镜像:
docker pull ghcr.io/inoxoft/whitelightning:latest
- 配置 OpenRouter API 密钥:
export OPEN_ROUTER_API_KEY="sk-..."
(仅训练阶段使用) - 运行命令生成模型,例如情感分析模型:
docker run --rm -e OPEN_ROUTER_API_KEY="sk-..." -v "$(pwd)":/app/models ghcr.io/inoxoft/whitelightning:latest python -m text_classifier.agent -p "Categorize customer reviews as positive, neutral, or negative"
工具会自动完成合成数据生成、模型训练和优化,10-15 分钟后即可获得小于 1MB 的 ONNX 模型,完全离线运行,支持多语言部署。
本答案来源于文章《WhiteLightning:一键生成轻量级离线文本分类模型的开源工具》