Files
ai-toolkit/toolkit/stable_diffusion_model.py