diff --git a/fastchat/model/model_adapter.py b/fastchat/model/model_adapter.py index 9aae91120..3277bbe1a 100644 --- a/fastchat/model/model_adapter.py +++ b/fastchat/model/model_adapter.py @@ -872,7 +872,11 @@ class OpenChat35Adapter(BaseModelAdapter): """The model adapter for OpenChat 3.5 (e.g. openchat/openchat_3.5)""" def match(self, model_path: str): - return "openchat" in model_path.lower() and "3.5" in model_path.lower() + if "openchat" in model_path.lower() and "3.5" in model_path.lower(): + return True + elif "starling-lm" in model_path.lower(): + return True + return False def get_default_conv_template(self, model_path: str) -> Conversation: return get_conv_template("openchat_3.5") diff --git a/fastchat/model/model_registry.py b/fastchat/model/model_registry.py index ed1e3337c..79573af26 100644 --- a/fastchat/model/model_registry.py +++ b/fastchat/model/model_registry.py @@ -188,6 +188,12 @@ def get_model_info(name: str) -> ModelInfo: "https://github.com/imoneoi/openchat", "OpenChat 3.5 is a versatile, open-source language model fine-tuned using C-RLFT", ) +register_model_info( + ["starling-lm-7b-alpha"], + "Starling-LM-7B-alpha", + "https://huggingface.co/berkeley-nest/Starling-LM-7B-alpha", + "an open model trained by RLAIF", +) register_model_info( ["llama-7b", "llama-13b"], "LLaMA",