add starling support

lm-sys · Nov 27, 2023 · 280da6d · 280da6d
1 parent 76fbdef
commit 280da6d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git 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
@@ -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",