Have you tried to train DiT with DCAE f64 and f128, or train UViT-2B with DCAE f128? Thanks, Yaobo