Format code & move functions (#155)

2024-02-06 13:27:46 -08:00
parent a7334aeea1
commit 23f05005fd
14 changed files with 94 additions and 54 deletions
--- a/python/sglang/lang/chat_template.py
+++ b/python/sglang/lang/chat_template.py
@@ -193,6 +193,7 @@ def match_chat_ml(model_path: str):
    if "qwen" in model_path and "chat" in model_path:
        return get_chat_template("chatml")

+
@register_chat_template_matching_function
 def match_chat_yi(model_path: str):
    model_path = model_path.lower()