FLUX.1作为高性能图像生成模型,对硬件和配置有严格要求:
- 显存要求:至少需要24GB显存的Nvidia GPU(如RTX 4090或专业级显卡),云端训练推荐使用A40(48GB显存)实例
- 特殊配置::
- existir
.env
文件中必须添加Hugging Face的HF_TOKEN
(需先接受模型许可协议) - 配置文件中需明确设置
quantize: true
responder cantandois_flux: true
parâmetros - 低显存环境下需启用
low_vram: true
在CPU上量化模型
- existir
- Seleção de modelos:需从Hugging Face下载特定版本(如FLUX.1-dev或FLUX.1-schnell),并搭配对应的训练适配器(如
ostris/FLUX.1-schnell-training-adapter
)
特别提示:训练过程中应避免在保存检查点时中断,否则可能造成文件损坏。建议初次使用者先参考项目提供的train_lora_flux_24gb.yaml
示例配置文件进行参数调整。
Essa resposta foi extraída do artigoKit de ferramentas de IA da Ostris: Difusão estável com o kit de ferramentas de treinamento de modelos FLUX.1O