Files
diffsynth-studio/diffsynth/diffusion/training_module.py