Files
modelscope/tests/trainers/test_custom_diffusion_trainer.py