Sharded version of the original https://huggingface.co/openlm-research/open_llama_3b_v2