FluxGym: GUI de entrenamiento con baja VRAM para Flux LoRA

¿Alguna vez quisiste entrenar FLUX LoRAs pero te quedaste atascado debido a los altos requisitos de VRAM? Presentamos Flux Gym.

Interfaz de usuario (UI)

¿De qué se trata Flux Gym?

Flux Gym es una interfaz web sencilla para entrenar modelos FLUX LoRA. ¿La mejor parte? Funciona con configuraciones de VRAM más bajas: hablamos de 12GB, 16GB y 20GB.

La cosa es la siguiente:

• Está basado en el proyecto AI-Toolkit (kudos a @multimodalart por la estilizada interfaz Gradio)
• Bajo el capó, usa Kohya Scripts para el entrenamiento real

¿Por qué Flux Gym?

Oye, lo entiendo. Entrenar FLUX LoRAs puede ser un problema:

• AI-Toolkit es excelente, pero necesita 24GB de VRAM. ¡Ay!
• Los Kohya Scripts son flexibles, pero ¿la línea de comandos? No es para todos.

Características principales

• Funciona con configuraciones de 12GB, 16GB y 20GB de VRAM
• Interfaz web extremadamente sencilla
• Impulsado por Kohya Scripts (por lo que es expandible)

Empezar

La forma fácil: Instalación con un clic

1. Ve a Pinokio: https://pinokio.computer/item?uri=https://github.com/cocktailpeanut/fluxgym
2. Haz clic, espera, listo. Configurará todo localmente.

La forma casera: Instalación manual

1. Clona los repositorios:

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

2. Configura un entorno virtual:

   • Windows:

     python -m venv env
     env\Scripts\activate
     

   • Linux:

     python -m venv env
     source env/bin/activate
     

3. Instala las dependencias:

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

4. Obtén PyTorch Nightly:

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

5. Descarga los checkpoints del modelo (consulta las instrucciones originales para los enlaces)

6. Arráncalo:

   python app.py
   

Cómo usar Flux Gym

Es sencillo:

1. Ingresa tu información de LoRA
2. Sube imágenes y agrega subtítulos (usa esa palabra desencadenante)
3. Pulsa "start" (iniciar)

Eso es todo. En serio.

Preguntas frecuentes (FAQ)

P: ¿Necesito conocimientos de programación para usar Flux Gym? R: No. La interfaz web lo hace tan fácil como hacer clic.

P: ¿Puedo usarlo en una GPU de laptop? R: Si tienes 12GB de VRAM o más, estás listo.

P: ¿Cómo se compara con otras herramientas de entrenamiento de FLUX LoRA? R: Es más sencillo de usar pero aún así tiene el poder de Kohya Scripts.

En resumen

FluxGym cierra la brecha entre la facilidad de uso y los requisitos bajos de VRAM para el entrenamiento de FLUX LoRA. No es elegante, no es complicado: simplemente funciona.

¿Quieres profundizar más? Consulta el hilo completo: https://x.com/cocktailpeanut/status/1832084951115972653

Pruébalo. Tu GPU te lo agradecerá.

Citaciones: https://github.com/cocktailpeanut/fluxgym