perfectgf/lora-dataset-studio
Self-hosted, one-tab workbench for the whole training lifecycle: build Character/Concept/Style datasets (generate, scrape, or triage a big dump in the Image Bank), curate, caption, clean watermarks, then train LoRAs on five model families or a full Krea 2 model on a rented GPU, and rank checkpoints in a Test Studio. Free & local.
它解决了什么问题
LoRA Dataset Studio 提供了一个统一的浏览器界面,涵盖为 LoRA 训练创建图像数据集的完整生命周期。它消除了在收集、清理、标注和训练图像时需要在多个不同工具之间切换的需求,简化了教导 AI 模型特定角色、概念或风格的过程。
它的运作方式
该软件作为一个中央枢纽,集成了多个专门的 AI 工具。它在用户的机器上本地运行,并利用 ai-toolkit 进行实际的训练过程,以及 ComfyUI 进行本地图像生成和检查点测试。它将工作流程组织成几个阶段:
- 数据集管理: 导入图像、将其组织成数据库,并通过裁剪、镜像和质量评分进行策展。
- 标注: 使用本地视觉模型或 JoyCaption 为图像生成描述或标签。
- 训练: 为各种模型系列(例如 SDXL、FLUX.1 和 Qwen-Image 2.1)提供引导式配方,并管理训练队列。
- 审查: 使用固定种子和提示批次比较不同的检查点,以评估训练后 LoRA 的质量。
适用对象
它专为 AI 艺术家和开发人员设计,他们希望为图像生成模型创建高质量的 LoRA,而无需管理复杂的命令行工具或手动文件组织。
亮点
- 端到端工作流程: 涵盖从图像库策展和文本检测到训练和最终测试的所有内容。
- 可扩展的插件系统: 提供可选的插件,用于云训练(通过 vast.ai)、网页抓取,以及发布到 Civitai 或 Hugging Face。
- 多模型支持: 支持多种训练系列,包括 FLUX.1、SDXL 和 Krea 2。
- 集成测试: 包含一个“工作室”,用于比较检查点和强度,并提供谱系和排名工具。
相关
- 项目
- 项目
- 项目
- 项目