feat: add also base models

2026-05-03 11:20:36 +00:00 · 2025-05-21 04:11:07 +04:00
parent 9e9575665e
commit f314d18863
1 changed files with 6 additions and 0 deletions
@@ -50,6 +50,12 @@ SUPPORTED_HF_MODELS = {
    "tiiuae/Falcon-E-1B-Instruct": {
        "model_name": "Falcon-E-1B-Instruct",
    },
+    "tiiuae/Falcon-E-3B-Base": {
+        "model_name": "Falcon-E-3B-Base",
+    },
+    "tiiuae/Falcon-E-1B-Base": {
+        "model_name": "Falcon-E-1B-Base",
+    },
 }

 SUPPORTED_QUANT_TYPES = {