From a73261dbb8d75216e74b129fc2919dc1205f4dc3 Mon Sep 17 00:00:00 2001
From: mitya <dimitry.ageev@gmail.com>
Date: Mon, 30 Oct 2023 13:23:25 +0300
Subject: [PATCH] refact model work better with spm spad

---
 src/caps.rs | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/src/caps.rs b/src/caps.rs
index b3a8fedc7..03e4315c6 100644
--- a/src/caps.rs
+++ b/src/caps.rs
@@ -57,7 +57,15 @@ const KNOWN_MODELS: &str = r#"
                 "FIM-SPM": {}
             },
             "default_scratchpad": "FIM-PSM",
-            "similar_models": ["bigcode/starcoderbase"]
+            "similar_models": [
+                "bigcode/starcoderbase",
+                "starcoder/15b/base",
+                "starcoder/15b/plus",
+                "starcoder/1b/base",
+                "starcoder/3b/base",
+                "starcoder/7b/base",
+                "wizardcoder/15b"
+            ]
         },
         "smallcloudai/Refact-1_6B-fim": {
             "n_ctx": 4096,
@@ -65,15 +73,9 @@ const KNOWN_MODELS: &str = r#"
                 "FIM-PSM": {},
                 "FIM-SPM": {}
             },
-            "default_scratchpad": "FIM-PSM",
+            "default_scratchpad": "FIM-SPM",
             "similar_models": [
-                "Refact/1.6B",
-                "starcoder/15b/base",
-                "starcoder/15b/plus",
-                "starcoder/1b/base",
-                "starcoder/3b/base",
-                "starcoder/7b/base",
-                "wizardcoder/15b"
+                "Refact/1.6B"
             ]
         },
         "codellama/CodeLlama-13b-hf": {