Format code (#275)

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>

train/process_ckpt.py

@@ -2,7 +2,7 @@ import torch, traceback, os, pdb
 from collections import OrderedDict
 
 
-def savee(ckpt, sr, if_f0, name, epoch,version):
+def savee(ckpt, sr, if_f0, name, epoch, version):
     try:
         opt = OrderedDict()
         opt["weight"] = {}
@@ -96,7 +96,7 @@ def show_info(path):
         return traceback.format_exc()
 
 
-def extract_small_model(path, name, sr, if_f0, info,version):
+def extract_small_model(path, name, sr, if_f0, info, version):
     try:
         ckpt = torch.load(path, map_location="cpu")
         if "model" in ckpt:
@@ -194,7 +194,7 @@ def change_info(path, info, name):
         return traceback.format_exc()
 
 
-def merge(path1, path2, alpha1, sr, f0, info, name,version):
+def merge(path1, path2, alpha1, sr, f0, info, name, version):
     try:
 
         def extract(ckpt):

train/utils.py

@@ -322,7 +322,11 @@ def get_hparams(init=True):
         "-sr", "--sample_rate", type=str, required=True, help="sample rate, 32k/40k/48k"
     )
     parser.add_argument(
-        "-sw", "--save_every_weights", type=str, default="0", help="save the extracted model in weights directory when saving checkpoints"
+        "-sw",
+        "--save_every_weights",
+        type=str,
+        default="0",
+        help="save the extracted model in weights directory when saving checkpoints",
     )
     parser.add_argument(
         "-v", "--version", type=str, required=True, help="model version"