Model complexity (#params or flops) comparison, and comparison to EDM2

[FutureXiang]

Dear authors,

I'm really curious about the efficiency of the proposed DiffiT models. It seems that another concurrent work from NVIDIA (by Karras), namely Analyzing and Improving the Training Dynamics of Diffusion Models, proposes EDM2, which is an improved version of ADM UNet, and achieves SOTA FIDs on ImageNet-512.

I really want to know what's the model complexity of DiffiT, and whether DiffiT (or EDM2) is the go-to choice with limited computational resources. It seems that the whole DiffiT paper doesn't contain any information about it.

Thank you!