A Modal-based application for training Stable Diffusion models using FluxGym interface.
- Runs on Modal cloud platform
- Uses NVIDIA CUDA for GPU acceleration
- Integrates with HuggingFace
- Web-based UI powered by Gradio
- Supports A100 GPU for training
- Modal account
- HuggingFace account and API token
- Python 3.11+
- Install Modal CLI:
pip install modal- Configure Modal:
modal token new- Set up HuggingFace secret in Modal:
modal secret create huggingface-secret HUGGINGFACE_SECRET=your_token_here- To run on modal:
modal serve app.py- CUDA Version: 12.4.0
- Base OS: Ubuntu 22.04
- Python Version: 3.11
- GPU: NVIDIA A100
The application uses Modal's serverless infrastructure to run a Gradio-based web interface for training FLUX models. It utilizes persistent volume storage for outputs and integrates with HuggingFace for model management.
MIT License
Contributions are welcome! Please feel free to submit a Pull Request.