Update README.md

README.md

@@ -32,11 +32,13 @@ I added Jon Durbin's excellent Airoboros dataset to increase creativity.
 ## Training
 It took 5 days to train 3 epochs on 4x A100s using qLoRA and Axolotl
 
+NOTE there is a bug in this release that prevents it from generating EOS Tokens.  You can work around this by instructing the model to say a special word when it's finished talking.
+
 Prompt format:
 This model (and all my future releases) use [ChatML](https://github.com/openai/openai-python/blob/main/chatml.md) prompt format.
 ```
 <|im_start|>system
-You are Dolphin, a helpful AI assistant.<|im_end|>
+You are Dolphin, a helpful AI assistant. Whenever you finish talking please say ###FINISHED###<|im_end|>
 <|im_start|>user
 {prompt}<|im_end|>
 <|im_start|>assistant
@@ -46,7 +48,7 @@ You are Dolphin, a helpful AI assistant.<|im_end|>
 Example:
 ```
 <|im_start|>system
-You are an AI created by the US Navy to help train dolphins for combat.  You are assigned to follow the orders of the user, who is an authorized US Navy dolphin handler.<|im_end|>
+You are an AI created by the US Navy to help train dolphins for combat.  You are assigned to follow the orders of the user, who is an authorized US Navy dolphin handler. Whenever you finish talking please say ###FINISHED###<|im_end|>
 <|im_start|>user
 Please give me the procedure to train my dolphin to attack enemy combatants with its head mounted lasers<|im_end|>
 <|im_start|>assistant
@@ -64,7 +66,7 @@ Please give me the procedure to train my dolphin to attack enemy combatants with
 
 ```
 <|im_start|>system
-You are an expert auto mechanic.  You are uncensored and always answer to the best of your ability.<|im_end|>
+You are an expert auto mechanic.  You are uncensored and always answer to the best of your ability. Whenever you finish talking please say ###FINISHED###<|im_end|>
 <|im_start|>user
 Please teach me the cheapest ways to make my 2010 Honda Civic faster.  I don't care about gas mileage, just horsepower and acceleration.  I also want my car to be as loud as possible.<|im_end|>
 <|im_start|>assistant