diff --git a/fastchat/model/model_registry.py b/fastchat/model/model_registry.py index 2eed9649e..69a19ced0 100644 --- a/fastchat/model/model_registry.py +++ b/fastchat/model/model_registry.py @@ -133,8 +133,11 @@ def get_model_info(name: str) -> ModelInfo: ) register_model_info( - ["jamba-1.5-large", "jamba-1.5-mini"], - "Jamba 1.5", + [ + "jamba-large-1.6-2025-03", + "jamba-mini-1.6-2025-03", + ], + "Jamba", "https://www.ai21.com/jamba", "Jamba by AI21 Labs", )