Flux 2: The shape of the latent argument is undocumented

[rolux]

The shape of the latent argument is not, as users might guess, (1, 16, h//8, w//8), but (1, 128, h//16, w//16).

This should be documented.

[preetam1407]

Hi! I’d like to work on this