From e2621f03959bc31e98c8604597300d0990ccb430 Mon Sep 17 00:00:00 2001 From: miri-bar <160584887+miri-bar@users.noreply.github.com> Date: Mon, 3 Mar 2025 15:30:35 +0200 Subject: [PATCH 1/2] feat: add new jamba models --- fastchat/model/model_registry.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/fastchat/model/model_registry.py b/fastchat/model/model_registry.py index 2eed9649e..1a3374d1e 100644 --- a/fastchat/model/model_registry.py +++ b/fastchat/model/model_registry.py @@ -133,8 +133,13 @@ def get_model_info(name: str) -> ModelInfo: ) register_model_info( - ["jamba-1.5-large", "jamba-1.5-mini"], - "Jamba 1.5", + [ + "jamba-1.5-large", + "jamba-1.5-mini", + "jamba-large-1.6-2025-03", + "jamba-mini-1.6-2025-03", + ], + "Jamba", "https://www.ai21.com/jamba", "Jamba by AI21 Labs", ) From 6e64a2b0f7b95ea32c193a3202d46a893b2f797d Mon Sep 17 00:00:00 2001 From: Miri Bar Date: Mon, 12 May 2025 12:03:48 +0300 Subject: [PATCH 2/2] feat: remove jamba 1.5 models --- fastchat/model/model_registry.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/fastchat/model/model_registry.py b/fastchat/model/model_registry.py index 1a3374d1e..69a19ced0 100644 --- a/fastchat/model/model_registry.py +++ b/fastchat/model/model_registry.py @@ -134,8 +134,6 @@ def get_model_info(name: str) -> ModelInfo: register_model_info( [ - "jamba-1.5-large", - "jamba-1.5-mini", "jamba-large-1.6-2025-03", "jamba-mini-1.6-2025-03", ],