# DiT from Wan2.1