sunxm2357's picture
Rename mp_rank_00_model_states.pt to diffusion.pt
27f5147 verified