Files
ai-toolkit/toolkit/util/qwen_vae_gradient_checkpointing.py