I was intrigued by "Mean Flows for One-Step Generative Modeling" and, in particular, how it handles averaging the marginal velocity field during training. In practice, they don't and replace it with the conditional one in their loss function. I wonder how mismatches impact generation...