Files
ai-toolkit/toolkit/models/mean_flow_adapter.py