Update README.md

README.md

@@ -5,6 +5,7 @@ license: apache-2.0
 tags:
 - text-generation
 - mlx
+- 4-bit
 datasets:
 - ehartford/wizard_vicuna_70k_unfiltered
 - totally-not-an-llm/EverythingLM-data-V3
@@ -12,118 +13,17 @@ datasets:
 - databricks/databricks-dolly-15k
 - THUDM/webglm-qa
 base_model: JackFram/llama-160m
-widget:
-- text: '<|im_start|>system
-
-    You are a helpful assistant, who answers with empathy.<|im_end|>
-
-    <|im_start|>user
-
-    Got a question for you!<|im_end|>
-
-    <|im_start|>assistant
-
-    Sure! What''s it?<|im_end|>
-
-    <|im_start|>user
-
-    Why do you love cats so much!? 🐈<|im_end|>
-
-    <|im_start|>assistant'
-- text: '<|im_start|>system
-
-    You are a helpful assistant who answers user''s questions with empathy.<|im_end|>
-
-    <|im_start|>user
-
-    Who is Mona Lisa?<|im_end|>
-
-    <|im_start|>assistant'
-- text: '<|im_start|>system
-
-    You are a helpful assistant who provides concise responses.<|im_end|>
-
-    <|im_start|>user
-
-    Heya!<|im_end|>
-
-    <|im_start|>assistant
-
-    Hi! How may I help you today?<|im_end|>
-
-    <|im_start|>user
-
-    I need to build a simple website. Where should I start learning about web development?<|im_end|>
-
-    <|im_start|>assistant'
-- text: '<|im_start|>user
-
-    Invited some friends to come home today. Give me some ideas for games to play
-    with them!<|im_end|>
-
-    <|im_start|>assistant'
-- text: '<|im_start|>system
-
-    You are a helpful assistant who answers user''s questions with details and curiosity.<|im_end|>
-
-    <|im_start|>user
-
-    What are some potential applications for quantum computing?<|im_end|>
-
-    <|im_start|>assistant'
-- text: '<|im_start|>system
-
-    You are a helpful assistant who gives creative responses.<|im_end|>
-
-    <|im_start|>user
-
-    Write the specs of a game about mages in a fantasy world.<|im_end|>
-
-    <|im_start|>assistant'
-- text: '<|im_start|>system
-
-    You are a helpful assistant who answers user''s questions with details.<|im_end|>
-
-    <|im_start|>user
-
-    Tell me about the pros and cons of social media.<|im_end|>
-
-    <|im_start|>assistant'
-- text: '<|im_start|>system
-
-    You are a helpful assistant who answers user''s questions with confidence.<|im_end|>
-
-    <|im_start|>user
-
-    What is a dog?<|im_end|>
-
-    <|im_start|>assistant
-
-    A dog is a four-legged, domesticated animal that is a member of the class Mammalia,
-    which includes all mammals. Dogs are known for their loyalty, playfulness, and
-    ability to be trained for various tasks. They are also used for hunting, herding,
-    and as service animals.<|im_end|>
-
-    <|im_start|>user
-
-    What is the color of an apple?<|im_end|>
-
-    <|im_start|>assistant'
-inference:
-  parameters:
-    max_new_tokens: 250
-    penalty_alpha: 0.5
-    top_k: 4
-    repetition_penalty: 1.01
 ---
 
 # Llama-160M-Chat-v1-4bit-mlx
-This model was converted to MLX format from [`Felladrin/Llama-160M-Chat-v1`]().
+This model was converted to MLX format from [`Felladrin/Llama-160M-Chat-v1`](https://huggingface.co/Felladrin/Llama-160M-Chat-v1).
 Refer to the [original model card](https://huggingface.co/Felladrin/Llama-160M-Chat-v1) 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/Llama-160M-Chat-v1-4bit-mlx --prompt "My name is"
+python generate.py --model mlx-community/Llama-160M-Chat-v1-4bit-mlx --prompt "<|im_start|>system\nYou are a helpful assistant who answers user's questions with details and curiosity.<|im_end|>\n<|im_start|>user\nWhat are some potential applications for quantum computing?<|im_end|>\n<|im_start|>assistant"
 ```