lint

lm-sys · Jan 16, 2025 · 135ab5d · 135ab5d
1 parent f7c0194
commit 135ab5d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/fastchat/conversation.py b/fastchat/conversation.py
@@ -2318,7 +2318,7 @@ def get_conv_template(name: str) -> Conversation:
         sep_style=SeparatorStyle.PHI4,
         sep="<|im_sep|>",
         stop_str="<|im_end|>",
-        stop_token_ids=[100257, 100265], # <|endoftext|>: 100257, <|im_end|>": 100265,
+        stop_token_ids=[100257, 100265],  # <|endoftext|>: 100257, <|im_end|>": 100265,
     )
 )
 

diff --git a/fastchat/model/model_registry.py b/fastchat/model/model_registry.py
@@ -255,9 +255,7 @@ def get_model_info(name: str) -> ModelInfo:
 )
 
 register_model_info(
-    [
-        "phi-4"
-    ],
+    ["phi-4"],
     "Phi-4",
     "https://techcommunity.microsoft.com/blog/aiplatformblog/introducing-phi-4-microsoft%E2%80%99s-newest-small-language-model-specializing-in-comple/4357090",
     "A capable and cost-effective small language model (SLMs) by Microsoft",