flux2 klein vae