Just a backup of ai-toolkit/flux2_vae