How to load a checkpoint model with SHARDED_STATE_DICT?

[Mustafa21]

I have found the solution for llama or you can edit for other models

you can use this :
llama-recipes/docs/inference.md at main · facebookresearch/llama-recipes · GitHub

i have edited some files inside it to load mistral instead of llama