Update on "Remove sharded ckpt from export_llama"

Sharded checkpoint isn't used anymore; removing it and simplifying export_llama.

Differential Revision: [D87828518](https://our.internmc.facebook.com/intern/diff/D87828518/)

[ghstack-poisoned]

Author: lucylq