Flux Gym : interface utilisateur simple pour l'entraînement FLUX LoRA avec une faible mémoire graphique

Introduction générale

Flux Gym是一个简易的网页用户界面，用于训练FLUX LoRA，支持低显存（12GB/16GB/20GB）。前端基于AI-Toolkit的Gradio UI，后端由Kohya Scripts提供支持。Flux Gym结合了AI-Toolkit WebUI的简易性和Kohya Scripts的灵活性，适用于多种显存配置，并支持Docker和自动下载模型功能。

Liste des fonctions

• 支持12GB、16GB、20GB显存
• Docker支持，自动下载模型
• 支持自定义基础模型
• 自动生成样本图像
• 发布到Huggingface
• 高级功能选项（隐藏）

Utiliser l'aide

Processus d'installation

一键安装

1. utiliserPinokio一键启动器自动安装并启动所有内容：Pinokio一键启动器

installation manuelle

1. 克隆Fluxgym和kohya-ss/sd-scripts：

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

2. Activer l'environnement virtuel :
   • Fenêtres.

     python -m venv env
     env\Scripts\activate
     

   • Linux.

     python -m venv env
     source env/bin/activate
     

3. Installer la dépendance :

   cd sd-scripts
   pip install -r requirements.txt
   cd ..
   pip install -r requirements.txt
   pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
   

4. Lancer l'application :

   python app.py
   

Installation de Docker

1. 克隆Fluxgym和kohya-ss/sd-scripts：

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

2. 构建镜像并运行：

   docker compose up -d --build
   

3. 打开浏览器并访问：http://localhost:7860

Fonction Opération Déroulement

1. 输入LoRA信息。
2. 上传图像并添加标签（使用触发词）。
3. 点击“开始”按钮。

样本图像配置

默认情况下，Fluxgym在训练期间不会生成样本图像。可以配置Fluxgym在每N步生成样本图像：

• 样本图像提示：这些提示将用于在训练期间自动生成图像。
• 样本图像每N步：例如，若“预期训练步数”为960，“样本图像每N步”为100，则图像将在第100、200、300步生成。

高级样本图像

使用内置的Kohya sd-scripts语法，可以完全控制训练阶段生成的样本图像：

• 触发词：例如，hrld person。
• 高级标志：例如，--d 标志指定种子，--w 指定图像宽度，--h 指定图像高度等。

发布到Huggingface

1. 获取Huggingface Token：Huggingface Token
2. 输入Token并点击“登录”。
3. 选择训练好的LoRA，编辑名称并发布到Huggingface。