Add missing "parameters" in "inference" property

README.md

@@ -6,51 +6,54 @@ base_model: Locutusque/TinyMistral-248M
 datasets:
   - OpenAssistant/oasst_top1_2023-08-25
 widget:
-- messages:
-  - role: user
-    content: "Invited some friends to come home today. Give me some ideas for games to play with them!"
-- messages:
-  - role: user
-    content: "How do meteorologists predict how much air pollution will be produced in the next year?"
-- messages:
-  - role: user
-    content: "Who is Mona Lisa?"
-- messages:
-  - role: user
-    content: "Heya!"
-  - role: assistant
-    content: "Hi! How may I help you today?"
-  - role: user
-    content: "I need to build a simple website. Where should I start learning about web development?"
-- messages:
-  - role: user
-    content: "What are some potential applications for quantum computing?"
-- messages:
-  - role: user
-    content: "Got a question for you!"
-  - role: assistant
-    content: "Sure! What's it?"
-  - role: user
-    content: "Why do you love cats so much!? 🐈"
-- messages:
-  - role: user
-    content: "Tell me about the pros and cons of social media."
-- messages:
-  - role: user
-    content: "Question: What is a dog?"
-  - role: assistant
-    content: "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."
-  - role: user
-    content: "Question: What is the capital of France?"
-  - role: assistant
-    content: "The capital of France is Paris. Paris is located in the north-central region of France and is known for its famous landmarks, such as the Eiffel Tower, the Louvre Museum, and Notre-Dame Cathedral."
-  - role: user
-    content: "Question: What is the color of an apple?"
+  - messages:
+    - role: user
+      content: "Invited some friends to come home today. Give me some ideas for games to play with them!"
+  - messages:
+    - role: user
+      content: "How do meteorologists predict how much air pollution will be produced in the next year?"
+  - messages:
+    - role: user
+      content: "Who is Mona Lisa?"
+  - messages:
+    - role: user
+      content: "Heya!"
+    - role: assistant
+      content: "Hi! How may I help you today?"
+    - role: user
+      content: "I need to build a simple website. Where should I start learning about web development?"
+  - messages:
+    - role: user
+      content: "What are some potential applications for quantum computing?"
+  - messages:
+    - role: user
+      content: "Got a question for you!"
+    - role: assistant
+      content: "Sure! What's it?"
+    - role: user
+      content: "Why do you love cats so much!? 🐈"
+  - messages:
+    - role: user
+      content: "Tell me about the pros and cons of social media."
+  - messages:
+    - role: user
+      content: "Question: What is a dog?"
+    - role: assistant
+      content: "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."
+    - role: user
+      content: "Question: What is the capital of France?"
+    - role: assistant
+      content: "The capital of France is Paris. Paris is located in the north-central region of France and is known for its famous landmarks, such as the Eiffel Tower, the Louvre Museum, and Notre-Dame Cathedral."
+    - role: user
+      content: "Question: What is the color of an apple?"
 inference:
-    max_new_tokens: 250,
-    penalty_alpha: 0.5,
-    top_k: 4,
+  parameters:
+    max_new_tokens: 250
+    penalty_alpha: 0.5
+    top_k: 5
     repetition_penalty: 1.001
+    stop:
+      - <|im_end|>
 ---
 
 # Locutusque's TinyMistral-248M trained on OpenAssistant TOP-1 Conversation Threads