Update README.md

README.md

@@ -16,6 +16,11 @@ The comparison can be found in [this wandb report](https://api.wandb.ai/links/em
 ![image/png](https://cdn-uploads.huggingface.co/production/uploads/6317aade83d8d2fd903192d9/NAw-hyWJl5ihVsAPqz3Xe.png)
 
 Sample inference code
+
+```sh
+pip install ai2-olmo
+```
+
 ```python
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline, TextStreamer