kohya_ss 简介
kohya_ss 是一款功能强大的 AI 绘画模型训练工具,它优化了传统模型训练的复杂流程,让普通用户也能轻松掌握 AI 绘画模型的训练技巧。无论是 Stable Diffusion 模型的微调,还是 LoRA、Textual Inversion 等技术的应用,kohya_ss 都提供了直观的操作界面和完善的文档支持,帮助用户快速实现从技术小白到创作大师的跨越。
快速入门:kohya_ss 的核心优势
kohya_ss 之所以能成为 AI 绘画爱好者的首选工具,源于其三大核心优势:
1. 零代码门槛的图形界面
通过 kohya_gui/ 目录下的图形界面工具,用户无需编写任何代码即可完成训练参数配置、数据准备和模型生成的全流程。这种可视化操作极大降低了技术门槛,让更多创作者能够专注于创意本身。
2. 全面支持主流训练技术
kohya_ss 支持目前 AI 绘画领域最流行的训练技术,包括:
- LoRA (Low-Rank Adaptation) 轻量化模型训练
- Textual Inversion 文本嵌入训练
- DreamBooth 个性化模型微调
- ControlNet 条件控制训练
这些技术都可以通过 presets/ 目录下的预设配置文件快速应用,无需深入理解复杂的算法原理。
3. 详尽的官方文档
项目提供了完善的中文文档,如 docs/train_README-zh.md,涵盖从安装到高级训练的所有细节。即使是完全没有 AI 背景的用户,也能按照文档逐步完成模型训练。
安装指南:三步开启你的 AI 创作之旅
环境准备
kohya_ss 提供了多种安装方式,适应不同操作系统和用户需求:
- Windows 用户:直接运行 setup.bat 脚本,自动配置所需环境
- Linux 用户:执行 setup.sh 进行安装
- RunPod 云端环境:使用 setup-runpod.sh 快速部署
对于追求效率的用户,推荐使用 UV 包管理器,通过 gui-uv.sh(Linux/Mac)或 gui-uv.bat(Windows)启动,可显著提升依赖安装速度。
启动图形界面
安装完成后,通过以下命令启动 kohya_ss 的图形界面:
python kohya_gui.py
或使用对应平台的启动脚本:gui.sh(Linux/Mac)、gui.bat(Windows)或 gui.ps1(PowerShell)。
数据准备:打造高质量训练数据集
训练优秀的 AI 绘画模型,高质量的数据集是基础。kohya_ss 推荐的文件组织结构如下:
images/
├── 30_cat/
│ ├── image1.jpg
│ ├── image1.txt
│ ├── image2.png
│ └── image2.txt
├── 30_dog/
│ └── ...
└── 40_black mamba/
└── ...
这种结构通过文件夹名称(如"30_cat")指定图像权重,数字越大表示该类别的重要性越高。每个图像文件对应一个文本文件,用于存储图像描述,如"Dariusz_Zawadzki.txt"中的内容将作为训练时的提示词。
小贴士:使用 tools/caption.py 工具可以批量生成图像描述,大幅减少人工标注工作量。

