Correct phi-1.5 prompt format

jncraton · Jan 4, 2024 · b34d350 · b34d350
1 parent d5271e1
commit b34d350
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/changelog.md b/changelog.md
@@ -8,6 +8,7 @@
 - Reduce memory usage for large document embeddings
 - Updated to TinyLlama Chat v1.0
 - Remove auto model scaling on Colab
+- Correct phi-1.5 prompt format
 
 ### Added
 

diff --git a/languagemodels/config.py b/languagemodels/config.py
@@ -198,6 +198,7 @@ class ModelFilterException(Exception):
         "backend": "ct2",
         "architecture": "decoder-only-transformer",
         "license": "other",
+        "prompt_fmt": "{instruction}\n\nAnswer:",
     },
     {
         "name": "LaMini-GPT-774M",