Files
ai-toolkit/toolkit/models/wan21/autoencoder_kl_wan.py