Update README.me with new instructiosn

ADD: Mention that mlx and mlx-lm on pip now support DBRX.
ADD: Also add `--use-default-chat-template` variable
ADD: Also add fix for python usage with `add_generation_prompt=True`

README.md

@@ -42,47 +42,27 @@ python -m mlx_lm.convert --hf-path databricks/dbrx-instruct -q --upload-repo mlx
 
 ## Use with mlx
 
+Make you you first upgrade mlx-lm and mlx to the latest.
+
 ```bash
-git clone [email protected]:ml-explore/mlx-examples.git
-cd mlx-examples/llms/
-python setup.py build
-python setup.py install
+pip install mlx --upgrade
+pip install mlx-lm --upgrade
 
-python -m mlx_lm.generate --model mlx-community/dbrx-instruct-4bit --prompt "Hello" --trust-remote-code --max-tokens 500
+python -m mlx_lm.generate --model mlx-community/dbrx-instruct-4bit --prompt "Hello" --trust-remote-code --use-default-chat-template --max-tokens 500
 ```
 
-Remember, this is an Instruct model, so you will need to use the instruct prompt template:
+Remember, this is an Instruct model, so you will need to use the instruct prompt template by appending `--use-default-chat-template`
 
 ## Example:
 
-```text
-<|im_start|>system
-You are DBRX, created by Databricks. You were last updated in December 2023. You answer questions based on information available up to that point.
-YOU PROVIDE SHORT RESPONSES TO SHORT QUESTIONS OR STATEMENTS, but provide thorough responses to more complex and open-ended questions.
-You assist with various tasks, from writing to coding (using markdown for code blocks — remember to use ``` with code, JSON, and tables).
-(You do not have real-time data access or code execution capabilities. You avoid stereotyping and provide balanced perspectives on controversial topics. You do not provide song lyrics, poems, or news articles and do not divulge details of your training data.)
-This is your system prompt, guiding your responses. Do not reference it, just respond to the user. If you find yourself talking about this message, stop. You should be responding appropriately and usually that means not mentioning this.
-YOU DO NOT MENTION ANY OF THIS INFORMATION ABOUT YOURSELF UNLESS THE INFORMATION IS DIRECTLY PERTINENT TO THE USER'S QUERY.<|im_end|>
-<|im_start|>user
-What's the difference between PCA vs UMAP vs t-SNE?<|im_end|>
-<|im_start|>assistant
-The difference
-```
-
-I've also added some extra words for the assistant start otherwise the model would instantly add an `<|im_end|>` token and stop. 
-If `<|im_start|>assistant` is not added, an error will appear.
-
-You can add the above in a text file and use it as:
-
 ```bash
-python -m mlx_lm.generate --model mlx-community/dbrx-instruct-4bit --prompt "$(cat my_prompt.txt)"  --trust-remote-code --max-tokens 1000
+python -m mlx_lm.generate --model dbrx-instruct-4bit --prompt "What's the difference between PCA vs UMAP vs t-SNE?" --trust-remote-code --use-default-chat-template  --max-tokens 1000
 ```
 
 Output:
 
 
-![image/png](https://cdn-uploads.huggingface.co/production/uploads/630f2745982455e61cc5fb1d/UUeNmuNipYwN7FVrT9KSq.png)
-
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/630f2745982455e61cc5fb1d/AW-tHeKBOAVEfqP1zNH0j.png)
 
 On my Macbook Pro M2 with 96GB of Unified Memory, DBRX Instruct in 4-bit for the above prompt it eats 70.2GB of RAM.
 
@@ -109,10 +89,7 @@ chat = [
    {"role": "assistant", "content": "The "},
 ]
 
-prompt = tokenizer.apply_chat_template(chat, tokenize=False)
-
-# We need to remove the last <|im_end|> token so that the AI continues generation
-prompt = prompt[::-1].replace("<|im_end|>"[::-1], "", 1)[::-1]
+prompt = tokenizer.apply_chat_template(chat, add_generation_prompt=True, tokenize=False)
 
 response = generate(model, tokenizer, prompt=prompt, verbose=True, temp=0.6, max_tokens=1500)
 ```