Flux Kontext Lora 模型训练环境搭建

发布于:2025-07-19 ⋅ 阅读:(12) ⋅ 点赞:(0)

🎈文章概述

Flux Kontext出现了有一段时间,kontext lora 其实与之前flux版本的lora并不直接通用,可以说flux的lora更多是教会Flux模型一些静态的知识和信息,而Kontext模型本身更多是需要教会模型“编辑”的动态信息。Kontext本身理解的图像编辑的信息有限,所以在一些场景下,需要通过Kontext对应的lora来强化某些“图像编辑”动作。

那么如何搭建Kontext Lora的训练环境呢?

网络上有一些资料,但大多比较复杂。这里就自己搭建环境的过程,简单写一个指南,供需要的朋友参考。

🎡操作步骤

【第一步】准备硬件

你需要准备一台显卡比较好的Nvidia显卡,最好是24G及以上显存(如果使用NF4量化模型,据说16G显存也可以,但是我没有尝试,大家可以自行尝试。)

同时,保证电脑空余硬盘20G以上,我安装完之后大概会是17.2G,后续还需要保存lora,所以最少也要20G以上。

【第二步】安装软件

现在使用比较多的训练器是https://github.com/lrzjason/T2ITrainer,可以支持量化模型,速度相对较快,配置要求相对较低。

打开目标硬盘的目标目录(路径建议不要有中文),然后在地址栏输入cmd并回车。

在命令行中输入(需要保持网络畅通):

git clone https://github.com/lrzjason/T2ITrainer.git

之后在目录里你会看到多了一个T2ITrainer的文件夹。

【第三步】设置虚拟环境

进入该文件夹,然后同样地址栏输入cmd并回车打开命令行。

建立虚拟环境。

python -m venv venv

接着输入以下命令激活虚拟环境

.\venv\Scripts\activate

类似这样:

【第四步】安装依赖

在打开的虚拟环境中,安装torch和torchvison。

分别输入以下命令后并回车。

pip install torch==2.7.0
pip install torchvision==0.22.0

之后安装其他相关依赖

pip install -r requirements.txt

有些朋友可能安装这一步会因为网络不太好容易出错,所以可以从国内镜像安装。

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

【第五步】下载模型

从以下链接下载模型:

https://pan.quark.cn/s/81d9d518f601

之后将整个文件夹,放在T2ITrainer文件夹的根目录下

 

【第六步】开始训练

在虚拟环境中输入以下指令启动训练界面的后台。

python ui_flux_fill.py

之后就可以复制这个url,到浏览器中手动打开:

接下来就可以配置参数和开始训练咯。

【第七步】后续训练

后续再使用的话,只需要先进入T2ITrainer目录,然后地址栏输入cmd进入终端。

执行:

.\venv\Scripts\activate

打开虚拟环境。

然后虚拟环境输入:

python ui_flux_fill.py

启动训练后台。

然后到浏览器打开虚拟环境中提示的网址,就可以愉快的开始训练kontext的lora啦~

最后,这里是“聚梦小课堂”,如果觉得不错,请点个赞支持一下👍


网站公告

今日签到

点亮在社区的每一天
去签到