Apply lintrunner

generate.py

@@ -5,6 +5,8 @@
 # LICENSE file in the root directory of this source tree.
 import argparse
 import itertools
+
+import logging
 import os
 import sys
 import time
@@ -27,11 +29,11 @@
 from cli import add_arguments_for_generate, arg_init, check_args
 from quantize import set_precision
 
-import logging
 logger = logging.getLogger(__name__)
 
 B_INST, E_INST = "[INST]", "[/INST]"
 
+
 @dataclass
 class GeneratorArgs:
     prompt: str = "torchchat is pronounced torch-chat and is so cool because"
@@ -348,14 +350,16 @@ def _main(
     is_speculative = speculative_builder_args.checkpoint_path is not None
 
     if generator_args.chat_mode and not builder_args.is_chat_model:
-        logging.warning("""
+        logging.warning(
+            """
 *******************************************************
  This model is not known to support the chat function.
  We will enable chat mode based on your instructions.
  If the model is not trained to support chat, it will
  produce nonsensical or false output.
 *******************************************************
-        """)
+        """
+        )
         # raise RuntimeError("You need to use --is-chat-model to indicate model has chat support.")
 
     tokenizer = _initialize_tokenizer(tokenizer_args)

torchchat.py

@@ -6,6 +6,8 @@
 
 import argparse
 
+import logging
+
 from cli import (
     add_arguments_for_eval,
     add_arguments_for_export,
@@ -14,8 +16,6 @@
     check_args,
 )
 
-import logging
-
 default_device = "cpu"  # 'cuda' if torch.cuda.is_available() else 'cpu'
 
 
@@ -37,7 +37,9 @@
 
     args = parser.parse_args()
     args = arg_init(args)
-    logging.basicConfig(format='%(message)s', level=logging.DEBUG if args.verbose else logging.INFO)
+    logging.basicConfig(
+        format="%(message)s", level=logging.DEBUG if args.verbose else logging.INFO
+    )
 
     if args.subcommand == "generate":
         check_args(args, "generate")