Bump the version to 0.2.35 (#2927)

lm-sys · Jan 17, 2024 · bb8aae5 · bb8aae5
1 parent d6ca36a
commit bb8aae5
Show file tree

Hide file tree

Showing 5 changed files with 28 additions and 11 deletions.
diff --git a/fastchat/__init__.py b/fastchat/__init__.py
@@ -1 +1 @@
-__version__ = "0.2.34"
+__version__ = "0.2.35"
diff --git a/fastchat/conversation.py b/fastchat/conversation.py
@@ -1434,9 +1434,9 @@ def get_conv_template(name: str) -> Conversation:
 )
 
 
-#yuan 2.0 template
-#reference:https://github.com/IEIT-Yuan/Yuan-2.0
-#reference:https://huggingface.co/IEITYuan
+# yuan 2.0 template
+# reference:https://github.com/IEIT-Yuan/Yuan-2.0
+# reference:https://huggingface.co/IEITYuan
 register_conv_template(
     Conversation(
         name="yuan",
@@ -1445,7 +1445,6 @@ def get_conv_template(name: str) -> Conversation:
         sep_style=SeparatorStyle.NO_COLON_SINGLE,
         sep="<sep>",
         stop_str="<eod>",
-
     )
 )
 

diff --git a/fastchat/model/model_adapter.py b/fastchat/model/model_adapter.py
@@ -2130,23 +2130,41 @@ def match(self, model_path: str):
     def get_default_conv_template(self, model_path: str) -> Conversation:
         return get_conv_template("solar")
 
+
 class Yuan2Adapter(BaseModelAdapter):
-    """The model adapter for Yuan """
+    """The model adapter for Yuan"""
 
     def match(self, model_path: str):
         return "yuan" in model_path.lower()
 
     def load_model(self, model_path: str, from_pretrained_kwargs: dict):
         model, tokenizer = super().load_model(model_path, from_pretrained_kwargs)
         tokenizer.add_tokens(
-            ['<sep>', '<pad>', '<mask>', '<predict>', '<FIM_SUFFIX>', '<FIM_PREFIX>', '<FIM_MIDDLE>', '<commit_before>',
-             '<commit_msg>', '<commit_after>', '<jupyter_start>', '<jupyter_text>', '<jupyter_code>',
-             '<jupyter_output>', '<empty_output>'], special_tokens=True)
+            [
+                "<sep>",
+                "<pad>",
+                "<mask>",
+                "<predict>",
+                "<FIM_SUFFIX>",
+                "<FIM_PREFIX>",
+                "<FIM_MIDDLE>",
+                "<commit_before>",
+                "<commit_msg>",
+                "<commit_after>",
+                "<jupyter_start>",
+                "<jupyter_text>",
+                "<jupyter_code>",
+                "<jupyter_output>",
+                "<empty_output>",
+            ],
+            special_tokens=True,
+        )
         return model, tokenizer
 
     def get_default_conv_template(self, model_path: str) -> Conversation:
         return get_conv_template("yuan")
 
+
 # Note: the registration order matters.
 # The one registered earlier has a higher matching priority.
 register_model_adapter(PeftModelAdapter)

diff --git a/fastchat/model/model_registry.py b/fastchat/model/model_registry.py
@@ -545,7 +545,7 @@ def get_model_info(name: str) -> ModelInfo:
 )
 
 register_model_info(
-    ["Yuan2-2B-hf","Yuan2-51B-hf","Yuan2-102B-hf"],
+    ["Yuan2-2B-hf", "Yuan2-51B-hf", "Yuan2-102B-hf"],
     "IEIYuan",
     "https://huggingface.co/IEITYuan",
     "Yuan2 is a Basemodel developed by IEI.",

diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "fschat"
-version = "0.2.34"
+version = "0.2.35"
 description = "An open platform for training, serving, and evaluating large language model based chatbots."
 readme = "README.md"
 requires-python = ">=3.8"