首页/新闻资讯/正文详情

kohya_ss:从 docker compose up 到第一个 LoRA 的 AI 模型训练实践

发布时间:2026/11/4 15:41:45 来源:云帆数科 栏目:资讯中心
kohya_ss:从 docker compose up 到第一个 LoRA 的 AI 模型训练实践
kohya_ss从 docker compose up 到第一个 LoRA 的 AI 模型训练实践【免费下载链接】kohya_ss项目地址: https://gitcode.com/GitHub_Trending/ko/kohya_sskohya_ss 是一个图形化的 AI 模型训练工具。一条 docker compose 命令即可拉起支持 SD 1.5、SDXL、SD3、Flux.1 等模型的训练环境覆盖 LoRA、Dreambooth、全量微调等用法。本文按从部署到跑通第一次训练的完整流程来讲。它能做的三件事训练一个风格 LoRA把一批统一风格的图放进训练集kohya_ss 会把这个风格学成几十 MB 大小的 LoRA 文件。之后在任何兼容底模里加载它就能生成同风格的新图。这是最常见的用法消费级显卡上一次训练通常不到一小时。个性化某个特定主体给同一角色 10~30 张照片它可以学到这张脸和标志性细节生成时用触发词就能呼出这个人。仓库自带一份测试数据集和配套的数据集配置放在 test/ 目录下可以先用它跑通一次完整训练来验证环境成本很低。微调不同的模型架构除了 LoRA还有全量微调finetuning和 Textual Inversion。底模覆盖 SD 1.5/2.x、SDXL、SD3、Flux.1、Lumina Image 2.0、Anima、HunyuanImage-2.1。当你要提升底模在某一类图上的整体表现或给模型词汇表加入全新概念时这两种方式比 LoRA 更合适。一条命令跑通 kohya_ss 部署开始前确认三件事已安装 Docker 并处于运行状态Windows 使用 WSL2 后端NVIDIA GPU 驱动正常nvidia-smi能看到显卡且已安装 NVIDIA Container Toolkit预留约 20GB 磁盘空间存放模型与缓存然后在项目根目录执行git clone --recursive https://gitcode.com/GitHub_Trending/ko/kohya_ss cd kohya_ss docker compose up -d--recursive不能省训练脚本是子模块漏掉会跑不起来。docker-compose.yaml 里定义了两个容器kohya-ss-gui 提供训练界面7860 端口tensorboard 提供训练曲线默认 6006 端口。浏览器不会自动打开手动访问 http://localhost:7860 看到带 LoRA、微调等页签的 kohya_ss 界面就算部署成功。数据方面把文件都放进dataset/目录图片放dataset/images训练日志和产出分别落在dataset/logs、dataset/outputs。compose 文件会把它们挂进容器容器随时可以销毁重建数据始终留在宿主机。完整跑一次 LoRA 训练以训练一个风格 LoRA 为例过程分四步。准备数据在dataset/images/下建一个专门文件夹放入 20~50 张同风格图片每张图旁边放同名.txt文件写一行描述。再写一个 TOML 格式的数据集配置指明图片文件夹image_dir、一个自造的触发词class_tokens例如 [V] fantasystyle、重复次数num_repeats和说明文件后缀。仓库里的 test/config/dataset.toml 就是一个可用示例照它的格式写即可。![kohya_ss AI 模型训练数据的单张图片示例需配同名 .txt 描述](https://raw.gitcode.com/GitHub_Trending/ko/kohya_ss/raw/f44226cfccca008094f958d829c49c74a7e9289d/test/img/10_darius kawasaki person/Dariusz_Zawadzki_4.jpg?utm_sourcegitcode_repo_files)选一组 LoRA 训练参数原则ranknetwork_dim决定网络容量从 16 起步看到过拟合样本图和训练图长得太像就调低 rank 与学习率欠拟合再往上加。学习率和优化器的组合因底模而异不建议硬记presets/lora/目录里有一批按 SD 1.5、SDXL 分开整理好的预设 JSON选一份就是稳妥的起点。分辨率按底模走SD 1.5 用 512SDXL 用 1024。启动训练在 GUI 的 LoRA 页签填入底模路径、数据集配置文件路径、网络参数和输出位置。注意容器内没有文件选择器路径需要手动输入dataset 目录在容器内挂载为/dataset所以填容器内路径。启动后训练过程中 GUI 会定期生成一批样本图。查看结果先看样本图风格对、且没有过拟合到只剩一种死样基本就可以停了。再打开 http://localhost:6006 的 TensorBoard 看损失曲线下降后趋平即可。训练出的 LoRA 文件位于宿主机dataset/outputs/可直接拿到 ComfyUI 或 WebUI 里使用。常见坑与提速办法页面打不开或训练报 GPU 错误说明容器没识别到显卡。先确认nvidia-smi正常、NVIDIA Container Toolkit 已安装再docker compose down docker compose up -d重建用docker compose logs kohya-ss-gui看具体报错。训练中途 OOMCUDA out of memory先降分辨率再减 batch_size还不够就开启梯度检查点用计算换显存。7860 或 6006 端口被占用改 docker-compose.yaml 里端口映射的左侧数字即可TensorBoard 端口也可以用.env里的 TENSORBOARD_PORT 覆盖。第一次训练慢、之后明显变快是因为首次需要把所有图编码成 latent开启 latent 缓存cache_latents会把它写入磁盘第二次训练直接跳过这一步。小贴士presets/lora/下的预设 JSON 是现成参数组自己调好的参数也可以存成预设放到presets/lora/user_presets/升级镜像用docker compose down docker compose up -d --pull always数据都在宿主机目录里不会丢从第一次训练开始建议先用 test/ 下的示例数据跑通一次完整训练确认环境端到端没问题再换成自己的数据。参数先用预设起步看到样本图之后再动 rank 和学习率。第一次的目标是产出一个能用的 LoRA而不是把每个参数都调到最优。【免费下载链接】kohya_ss项目地址: https://gitcode.com/GitHub_Trending/ko/kohya_ss创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

相关推荐

如何在ComfyUI中快速部署WanVideo FP8量化模型:5步实现高效视频生成
如何在ComfyUI中快速部署WanVideo FP8量化模型:5步实现高效视频生成

如何在ComfyUI中快速部署WanVideo FP8量化模型:5步实现高效视频生成 【免费下载链接】WanVideo_comfy_fp8_scaled 项目地址: https://ai.gitcode.com/hf_mirrors/Kijai/WanVideo_comfy_fp8_scaled WanVideo_comfy_fp8_scaled项目为AI视频创作者提供了一个完… · 2026/11/4 15:41:45

ok-ww 鸣潮后台刷声骸与一键日常快速指南
ok-ww 鸣潮后台刷声骸与一键日常快速指南

ok-ww 鸣潮后台刷声骸与一键日常快速指南 【免费下载链接】ok-wuthering-waves 鸣潮 后台自动战斗 自动刷声骸 一键日常 Automation for Wuthering Waves 项目地址: https://gitcode.com/GitHub_Trending/ok/ok-wuthering-waves ok-ww 是一个面向《鸣潮》的开源自动化工… · 2026/11/4 15:40:43

阅读APP书源导入实操指南:22个书源一次配好,10分钟开始阅读
阅读APP书源导入实操指南:22个书源一次配好,10分钟开始阅读

阅读APP书源导入实操指南:22个书源一次配好,10分钟开始阅读 【免费下载链接】Yuedu 📚「阅读」自用书源分享 项目地址: https://gitcode.com/gh_mirrors/yu/Yuedu 阅读APP装好并不能直接取书,第一步是导入书源。本文带你用… · 2026/11/4 15:40:43

MTP 推测解码 vs Mobius 知识-推理解耦:书生 S2 家族的两条提速路线,哪条更香
MTP 推测解码 vs Mobius 知识-推理解耦:书生 S2 家族的两条提速路线,哪条更香

MTP 推测解码 vs Mobius 知识-推理解耦:书生 S2 家族的两条提速路线,哪条更香 【免费下载链接】Intern-S2-397B 项目地址: https://ai.gitcode.com/InternLM/Intern-S2-397B 自研模型的军备竞赛走到今天,单纯堆参数已经不再是最性感的… · 2026/11/4 16:13:00

class-validator 验证的5个开关:如何让错误只出现在该出现的地方
class-validator 验证的5个开关:如何让错误只出现在该出现的地方

class-validator 验证的5个开关:如何让错误只出现在该出现的地方 【免费下载链接】class-validator Decorator-based property validation for classes. 项目地址: https://gitcode.com/gh_mirrors/cl/class-validator 更新用户资料的接口,有时 a… · 2026/11/4 16:11:59

March7thAssistant 上手指南:3 步跑通星穹铁道日常自动化
March7thAssistant 上手指南:3 步跑通星穹铁道日常自动化

March7thAssistant 上手指南:3 步跑通星穹铁道日常自动化 【免费下载链接】March7thAssistant 日常交给三月七,开拓留给你。《崩坏:星穹铁道》开源自动化助手。 项目地址: https://gitcode.com/gh_mirrors/ma/March7thAssistant March… · 2026/11/4 16:11:59

Boxer CLI参数速查表:从阈值到精度,所有命令行选项一次讲透
Boxer CLI参数速查表:从阈值到精度,所有命令行选项一次讲透

【免费下载链接】boxer Code for the Boxer research paper 项目地址: https://gitcode.com/gh_mirrors/boxer13/boxer 点击查看 免费下载 Boxer 是一个将开放世界 2D 检测框"抬升"(lifting)为 3D 方向包围盒(OBB&… · 2026/11/4 16:11:59

REA:本地 AI 逆向工程指南,用 1 条命令看清竞品应用的内部实现
REA:本地 AI 逆向工程指南,用 1 条命令看清竞品应用的内部实现

REA:本地 AI 逆向工程指南,用 1 条命令看清竞品应用的内部实现 【免费下载链接】rea Reverse engineer anything with agents, from app behavior down to native binaries. 项目地址: https://gitcode.com/GitHub_Trending/rea2/rea 你在别人的 … · 2026/11/4 16:11:59

chinese-xinhua 完整指南:免费构建中文语言知识库的终极方案
chinese-xinhua 完整指南:免费构建中文语言知识库的终极方案

chinese-xinhua 完整指南:免费构建中文语言知识库的终极方案 【免费下载链接】chinese-xinhua :orange_book: 中华新华字典数据库。包括歇后语,成语,词语,汉字。 项目地址: https://gitcode.com/gh_mirrors/ch/chinese-xinhua … · 2026/11/4 16:10:58

双极步进电机控制实战:DRV8818PWPR与PIC18F86K22驱动调试
双极步进电机控制实战:DRV8818PWPR与PIC18F86K22驱动调试

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views … · 2026/11/4 3:16:49

STM32F207ZG + DRV8818PWPR 步进电机驱动实战解析
STM32F207ZG + DRV8818PWPR 步进电机驱动实战解析

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views … · 2026/11/4 6:59:29

DRV8818PWPR+dsPIC30F4013工业级双极步进电机控制方案
DRV8818PWPR+dsPIC30F4013工业级双极步进电机控制方案

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views … · 2026/11/4 7:02:38

AI辅助技术栈评估:提升研发决策效率60%
AI辅助技术栈评估:提升研发决策效率60%

1. 项目背景与核心价值技术决策一直是研发团队面临的重大挑战。随着技术生态的快速演进,技术栈的选择、维护和迭代变得愈发复杂。我们团队在过去两年里,先后经历了三次重大技术栈迁移,每次都需要投入大量人力进行技术评估和风险分析。直到去年… · 2026/11/4 0:00:17

从生成Token到直接决策:Jev如何重塑AI Agent工程实践
从生成Token到直接决策:Jev如何重塑AI Agent工程实践

1. 从"生成文字"到"拍板决策":Jev 到底在解决什么问题大多数人第一次接触大模型,脑子里建立的模型是"输入问题、输出文字"。你问它今天天气怎么样,它给你一段描述;你让它写个周报,它给你… · 2026/11/4 0:00:17

32G统一内存跑大模型:内存账、TPS实测与端云分流
32G统一内存跑大模型:内存账、TPS实测与端云分流

想在一台 32G 统一内存的小主机上把大模型跑成"日常可用",这件事我前后折腾了小半年。标题里说的这台 Mac mini M6,是我们这次讨论的目标平台,具体到手规格以实物为准;但下面这套推算方法和实测思路,对任何统… · 2026/11/4 0:00:17

双极步进电机控制实战:DRV8818PWPR与PIC18F86K22驱动调试
双极步进电机控制实战:DRV8818PWPR与PIC18F86K22驱动调试

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views … · 2026/11/4 3:16:49

STM32F207ZG + DRV8818PWPR 步进电机驱动实战解析
STM32F207ZG + DRV8818PWPR 步进电机驱动实战解析

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views … · 2026/11/4 6:59:29

DRV8818PWPR+dsPIC30F4013工业级双极步进电机控制方案
DRV8818PWPR+dsPIC30F4013工业级双极步进电机控制方案

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views … · 2026/11/4 7:02:38

了解更多?预约专属演示

我们的顾问将为您一对一讲解产品与方案

企业微信二维码