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",