update a new sota model on MT-Bench which touch an 8.8 scores. (lm-sy…

…s#2864)
silk55 · Jan 26, 2024 · 4bc5d22 · 4bc5d22
1 parent f616540
commit 4bc5d22
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/fastchat/conversation.py b/fastchat/conversation.py
@@ -1323,7 +1323,7 @@ def get_conv_template(name: str) -> Conversation:
     )
 )
 # xDAN default template
-# source: https://huggingface.co/xDAN-AI/xDAN-L1-Chat-v0.1
+# source: https://huggingface.co/xDAN-AI/xDAN-L1-Chat-RL-v1
 register_conv_template(
     Conversation(
         name="xdan-v1",

diff --git a/fastchat/model/model_adapter.py b/fastchat/model/model_adapter.py
@@ -2032,10 +2032,10 @@ def get_default_conv_template(self, model_path: str) -> Conversation:
 
 
 class XdanAdapter(BaseModelAdapter):
-    """The model adapter for xDAN-AI (e.g. xDAN-AI/xDAN-L1-Chat-v0.1)"""
+    """The model adapter for xDAN-AI (e.g. xDAN-AI/xDAN-L1-Chat-RL-v1)"""
 
     def match(self, model_path: str):
-        return "xdan" in model_path.lower()
+        return "xdan" in model_path.lower() and "v1" in model_path.lower()
 
     def get_default_conv_template(self, model_path: str) -> Conversation:
         return get_conv_template("xdan-v1")

diff --git a/fastchat/model/model_registry.py b/fastchat/model/model_registry.py
@@ -510,9 +510,9 @@ def get_model_info(name: str) -> ModelInfo:
 )
 
 register_model_info(
-    ["xDAN-L1-Chat-v0.1"],
+    ["xDAN-L1-Chat-RL-v1"],
     "xDAN-L1-Chat",
-    "https://huggingface.co/xDAN-AI/xDAN-L1-Chat-v0.1",
+    "https://huggingface.co/xDAN-AI/xDAN-L1-Chat-RL-v1",
     "A large language chat model created by xDAN-AI.",
 )