update

README.md

@@ -7,6 +7,15 @@ license: bsd-3-clause
 
 This model was sharded using torch.float16. Use the code below to load this model, configure the device_map for your GPU/CPU split.
 
+First pull the model.
+
+```bash
+git clone https://huggingface.co/abacaj/codegen-16B-nl-sharded
+cd codegen-16B-nl-sharded
+git-lfs install
+git pull
+```
+
 ```python
 def load_model_sharded():
     config = AutoConfig.from_pretrained("abacaj/codegen-16B-nl-sharded")
@@ -27,7 +36,7 @@ def load_model_sharded():
     model = load_checkpoint_and_dispatch(
         model,
         dtype=torch.float16,
-        checkpoint="sharded",
+        checkpoint="codegen-16B-nl-sharded",
         device_map=device_map,
     ).eval()