Update README.md

README.md

@@ -9,13 +9,18 @@ datasets:
 - bigcode/starcoderdata
 ---
 
-# TinyLlama-1.1B-intermediate-step-1431k-3T-mlx
+# mlx-community/TinyLlama-1.1B-intermediate-step-1431k-3T-mlx
 This model was converted to MLX format from [`TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T`]().
 Refer to the [original model card](https://huggingface.co/TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T) for more details on the model.
 ## Use with mlx
+
 ```bash
-pip install mlx
-git clone https://github.com/ml-explore/mlx-examples.git
-cd mlx-examples/llms/hf_llm
-python generate.py --model mlx-community/TinyLlama-1.1B-intermediate-step-1431k-3T-mlx --prompt "My name is"
+pip install mlx-lm
+```
+
+```python
+from mlx_lm import load, generate
+
+model, tokenizer = load("mlx-community/TinyLlama-1.1B-intermediate-step-1431k-3T-mlx")
+response = generate(model, tokenizer, prompt="hello", verbose=True)
 ```