Add phi-2 prompt format

jncraton · Jan 4, 2024 · d5271e1 · d5271e1
1 parent 124fa4c
commit d5271e1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/languagemodels/config.py b/languagemodels/config.py
@@ -187,6 +187,7 @@ class ModelFilterException(Exception):
         "backend": "ct2",
         "architecture": "decoder-only-transformer",
         "license": "microsoft-research-license",
+        "prompt_fmt": "Instruct: {instruction}\nOutput:",
     },
     {
         "name": "phi-1_5",