600M-512px for comfyui not working

[Pancat009]

I download the Sana_600M_512px model in /checkpoint,
and download the whole Sana_600M_512px_diffusers in /models/diffusers
and copy the vae from /Sana_600M_512px_diffusers/vae to models/vae
but it repots error

Error(s) in loading state_dict for SanaMS:
size mismatch for pos_embed: copying a param with shape torch.Size([1, 256, 1152]) from checkpoint, the shape in current model is torch.Size([1, 1024, 1152]).