FluxGym：适用于Flux LoRA的低显存训练图形用户界面

是否曾经想训练FLUX LoRA，却因显存要求过高而受阻？那就来了解一下Flux Gym吧。

界面

Flux Gym是什么？

Flux Gym是一个简洁实用的网页界面，专为训练FLUX LoRA模型设计。最棒的是，它适用于更低显存的设备——比如12GB、16GB和20GB显存的设备都能运行。

具体来说：

• 它基于AI-Toolkit项目开发（特别感谢@multimodalart提供的流畅Gradio界面）
• 底层由Kohya脚本驱动，负责实际的训练工作

为什么选择Flux Gym？

我懂的，训练FLUX LoRA可能很麻烦：

• AI-Toolkit虽好，但需要24GB显存。这可有点让人头疼。
• Kohya脚本很灵活，但命令行操作？不是每个人都喜欢。

核心功能

• 支持12GB、16GB和20GB显存的设备
• 极其简单的网页界面
• 基于Kohya脚本（因此具有可扩展性）

开始使用

简单方式：一键安装

1. 访问Pinokio：https://pinokio.computer/item?uri=https://github.com/cocktailpeanut/fluxgym
2. 点击、等待，搞定。它会自动在本地完成所有设置。

手动方式：自行安装

1. 克隆仓库：

   git clone https://github.com/cocktailpeanut/fluxgym
   cd fluxgym
   git clone -b sd3 https://github.com/kohya-ss/sd-scripts
   

2. 创建虚拟环境：

   • Windows系统：

     python -m venv env
     env\Scripts\activate
     

   • Linux系统：

     python -m venv env
     source env/bin/activate
     

3. 安装依赖：

   cd sd-scripts
   pip install -r requirements.txt
   cd ..
   pip install -r requirements.txt
   

4. 获取PyTorch nightly版本：

   pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121
   

5. 下载模型检查点（具体链接请查看原始说明）

6. 启动程序：

   python app.py
   

如何使用Flux Gym

操作很简单：

1. 输入你的LoRA信息
2. 上传图片并添加描述（记得使用触发词）
3. 点击“开始”

真的就这么简单。

常见问题

问：使用Flux Gym需要编程技能吗？
答：不需要。网页界面支持纯点击操作，非常简单。

问：可以在笔记本电脑的GPU上使用吗？
答：如果你的显存至少有12GB，就可以使用。

问：与其他FLUX LoRA训练工具相比，它有什么优势？
答：它更易于使用，同时还具备Kohya脚本的强大功能。

总结

Flux Gym在FLUX LoRA训练的易用性和低显存需求之间架起了桥梁。它不花哨，也不复杂——但就是好用。

想深入了解？查看完整内容：https://x.com/cocktailpeanut/status/1832084951115972653

试试看吧。你的GPU会感谢你的。

参考资料： https://github.com/cocktailpeanut/fluxgym