format style

src/llmtuner/train/tuner.py

@@ -1,16 +1,18 @@
-import torch
 from typing import TYPE_CHECKING, Any, Dict, List, Optional
+
+import torch
 from transformers import PreTrainedModel
 
 from ..extras.callbacks import LogCallback
 from ..extras.logging import get_logger
-from ..hparams import get_train_args, get_infer_args
+from ..hparams import get_infer_args, get_train_args
 from ..model import load_model_and_tokenizer
-from .pt import run_pt
-from .sft import run_sft
-from .rm import run_rm
-from .ppo import run_ppo
 from .dpo import run_dpo
+from .ppo import run_ppo
+from .pt import run_pt
+from .rm import run_rm
+from .sft import run_sft
+
 
 if TYPE_CHECKING:
     from transformers import TrainerCallback
@@ -64,23 +66,23 @@ def export_model(args: Optional[Dict[str, Any]] = None):
     model.save_pretrained(
         save_directory=model_args.export_dir,
         max_shard_size="{}GB".format(model_args.export_size),
-        safe_serialization=(not model_args.export_legacy_format)
+        safe_serialization=(not model_args.export_legacy_format),
     )
     if model_args.export_hub_model_id is not None:
         model.push_to_hub(
             model_args.export_hub_model_id,
             token=model_args.hf_hub_token,
             max_shard_size="{}GB".format(model_args.export_size),
-            safe_serialization=(not model_args.export_legacy_format)
+            safe_serialization=(not model_args.export_legacy_format),
         )
 
     try:
-        tokenizer.padding_side = "left" # restore padding side
+        tokenizer.padding_side = "left"  # restore padding side
         tokenizer.init_kwargs["padding_side"] = "left"
         tokenizer.save_pretrained(model_args.export_dir)
         if model_args.export_hub_model_id is not None:
             tokenizer.push_to_hub(model_args.export_hub_model_id, token=model_args.hf_hub_token)
-    except:
+    except Exception:
         logger.warning("Cannot save tokenizer, please copy the files manually.")