From 0ebb58a11cb007c4c50f2c71d65b138a32b932c5 Mon Sep 17 00:00:00 2001
From: miri-bar <160584887+miri-bar@users.noreply.github.com>
Date: Tue, 25 Feb 2025 11:50:16 +0200
Subject: [PATCH] feat: add jamba feb to model registry, edit jamba names

---
 fastchat/model/model_registry.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/fastchat/model/model_registry.py b/fastchat/model/model_registry.py
index 2eed9649e..12cfb84bf 100644
--- a/fastchat/model/model_registry.py
+++ b/fastchat/model/model_registry.py
@@ -133,7 +133,12 @@ def get_model_info(name: str) -> ModelInfo:
 )
 
 register_model_info(
-    ["jamba-1.5-large", "jamba-1.5-mini"],
+    [
+        "jamba-1.5-large-2024-08",
+        "jamba-1.5-mini-2024-08",
+        "jamba-1.5-large-2025-02",
+        "jamba-1.5-mini-2025-02",
+    ],
     "Jamba 1.5",
     "https://www.ai21.com/jamba",
     "Jamba by AI21 Labs",