Update TinyLlama chat version

jncraton · Jan 3, 2024 · 3610958 · 3610958
1 parent d713b52
commit 3610958
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/changelog.md b/changelog.md
@@ -6,6 +6,7 @@
 
 - Improved search speed when searching many documents
 - Reduce memory usage for large document embeddings
+- Updated to TinyLlama Chat v1.0
 
 ### Added
 

diff --git a/languagemodels/config.py b/languagemodels/config.py
@@ -219,7 +219,7 @@ class ModelFilterException(Exception):
         ),
     },
     {
-        "name": "TinyLlama-1.1B-Chat-v0.3",
+        "name": "TinyLlama-1.1B-Chat-v1.0",
         "tuning": "instruct",
         "datasets": ["slimpajama", "starcoderdata"],
         "params": 1.1e9,
@@ -228,9 +228,8 @@ class ModelFilterException(Exception):
         "architecture": "decoder-only-transformer",
         "license": "mit",
         "prompt_fmt": (
-            "<|im_start|>user\n{instruction} <|im_end|> <|im_start|>assistant\n"
+            "<|user|>{instruction}<|assistant|>"
         ),
-        "support": "deprecated",
     },
     {
         "name": "codet5p-770m-py",