原生Win下的kohya_ss安装(附整合包)
介绍: 
1). kohya_ss可以用于训练lora 
2). 本次安装以Windows 11 专业版22631.3593为例,显卡驱动5445;部署版本为v24.1.4 
3). 建议在网络环境良好的情况下部署,需要开魔法; 
4). 自行部署需要确保安装了git和python3.10.x(建议3.10.11),并已经添加了对应的环境变量(建议看网盘里的部署文档); 
 
安装步骤: 
找一个文件夹(建议全英文路径),在地址栏输入cmd,按回车打开命令行窗口 
然后依次复制以下命令到终端并按回车; 
 
git clone --recurse-submodules https://github.com/bmaltais/kohya_ss.git 
 
cd kohya_ss 
 
python -m venv venv 
 
.\venv\Scripts\Activate.bat 
 
python.exe -m pip install --upgrade pip 
 
 
pip install -r requirements_windows.txt 
 
5).不要关闭终端窗口,环境部署完毕后需要进行加速设置,输入以下命令 
 
 
6).选择第5项"(Optional) Manually configure accelerate"然后回车,并按以下选项进行选择: 
*This machine 
*No distributed training 
no 
yes 
no 
no 
all 
bf16 
 
最后关闭窗口即可 
 
7). 打开记事本,复制添加以下内容: 
 
@echo off  
chcp 65001  
 
call venv\\Scripts\\activate.bat 
python kohya_gui.py --use-ipex --inbrowser 
 
@echo 启动完毕,请按任意键关闭  
call pause 
 
8). 文件保存为.bat格式,并放在kohya_ss根目录下,之后每次运行只要双击该文件,就能看到浏览器的lora训练界面,需要中文界面可在bat文件中加上--language zh-CN启动命令 
 
9). i卡的lora训练需要注意的几个点: 
a).LoRA—Training—Accelerate launch—Resource Selection选项卡需要将Mixed precision设置为bf16,同时Optimizer不能选用任何带8bit的选项; 
b).LoRA—Training—Paramenters—Advanced选项卡需要将CrossAttention设置为sdpa; 
c).a770 16G可直接训练768x768分辨率,a750训练768x768建议打开Memory efficient attention(内存高效注意力)选项,位于LoRA—Training—Paramenters—Advanced选项卡内; 
 
10). 附上整合包,需要自取 
提取码:q2mv 
 
2024.05.29 
由于之前的帖子被频道吞了,因此重新发布。 
 
 
测试截图 
 
参考资料 
 
2024-05-29
浏览990
LoRA训练
登录后评论
评论
9