Files
ai-toolkit/toolkit/models/v2/diffusion_models/qwen_image.py