formatting for linter

mkeoliya · Jun 10, 2023 · ba8bf40 · ba8bf40
1 parent c0bea31
commit ba8bf40
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/scripts/finetune.py b/scripts/finetune.py
@@ -64,11 +64,7 @@ def get_multi_line_input() -> Optional[str]:
 
 
 def do_inference(cfg, model, tokenizer, prompter="AlpacaPrompter"):
-    default_tokens = {
-        "unk_token": "<unk>",
-        "bos_token": "<s>",
-        "eos_token": "</s>"
-    }
+    default_tokens = {"unk_token": "<unk>", "bos_token": "<s>", "eos_token": "</s>"}
 
     for token, symbol in default_tokens.items():
         # If the token isn't already specified in the config, add it