fix a bug in cmd arguments about custom checkpoint and config

Author: Plachtaa

README-JA.md

@@ -91,7 +91,7 @@ python app.py
 
 リアルタイム音声変換GUI：
 ```bash
-python real-time-gui.py --checkpoint <path-to-checkpoint> --config <path-to-config>
+python real-time-gui.py --checkpoint-path <path-to-checkpoint> --config-path <path-to-config>
 ```
 - `checkpoint` は独自のモデルをトレーニングまたはファインチューニングした場合のモデルチェックポイントへのパス、空白の場合はhuggingfaceからデフォルトモデルを自動ダウンロード（`seed-uvit-tat-xlsr-tiny`）
 - `config` は独自のモデルをトレーニングまたはファインチューニングした場合のモデル設定へのパス、空白の場合はhuggingfaceからデフォルト設定を自動ダウンロード

README-ZH.md

@@ -81,7 +81,7 @@ python app.py
 
 实时语音转换 GUI:
 ```bash
-python real-time-gui.py --checkpoint <path-to-checkpoint> --config <path-to-config>
+python real-time-gui.py --checkpoint-path <path-to-checkpoint> --config-path <path-to-config>
 ```
 - `checkpoint` 模型检查点路径，若为空将自动下载默认模型 (`seed-uvit-tat-xlsr-tiny`)
 - `config` 模型配置文件路径，若为空将自动下载默认配置  

README.md

@@ -94,7 +94,7 @@ This will only load pretrained models for zero-shot inference. To use custom che
 
 Real-time voice conversion GUI:
 ```bash
-python real-time-gui.py --checkpoint <path-to-checkpoint> --config <path-to-config>
+python real-time-gui.py --checkpoint-path <path-to-checkpoint> --config-path <path-to-config>
 ```
 - `checkpoint` is the path to the model checkpoint if you have trained or fine-tuned your own model, leave to blank to auto-download default model from huggingface. (`seed-uvit-tat-xlsr-tiny`)
 - `config` is the path to the model config if you have trained or fine-tuned your own model, leave to blank to auto-download default config from huggingface  

app_svc.py

@@ -20,14 +20,14 @@ def load_models(args):
     print(f"Using device: {device}")
     print(f"Using fp16: {fp16}")
     # f0 conditioned model
-    if args.checkpoint_path is None or args.checkpoint_path == "":
+    if args.checkpoint is None or args.checkpoint == "":
         dit_checkpoint_path, dit_config_path = load_custom_model_from_hf("Plachta/Seed-VC",
                                                                          "DiT_seed_v2_uvit_whisper_base_f0_44k_bigvgan_pruned_ft_ema_v2.pth",
                                                                          "config_dit_mel_seed_uvit_whisper_base_f0_44k.yml")
     else:
-        print(f"Using custom checkpoint: {args.checkpoint_path}")
-        dit_checkpoint_path = args.checkpoint_path
-        dit_config_path = args.config_path
+        print(f"Using custom checkpoint: {args.checkpoint}")
+        dit_checkpoint_path = args.checkpoint
+        dit_config_path = args.config
     config = yaml.safe_load(open(dit_config_path, "r"))
     model_params = recursive_munch(config["model_params"])
     model_params.dit_type = 'DiT'

app_vc.py

@@ -19,13 +19,13 @@ def load_models(args):
     fp16 = args.fp16
     print(f"Using device: {device}")
     print(f"Using fp16: {fp16}")
-    if args.checkpoint_path is None or args.checkpoint_path == "":
+    if args.checkpoint is None or args.checkpoint == "":
         dit_checkpoint_path, dit_config_path = load_custom_model_from_hf("Plachta/Seed-VC",
                                                                          "DiT_seed_v2_uvit_whisper_small_wavenet_bigvgan_pruned.pth",
                                                                          "config_dit_mel_seed_uvit_whisper_small_wavenet.yml")
     else:
-        dit_checkpoint_path = args.checkpoint_path
-        dit_config_path = args.config_path
+        dit_checkpoint_path = args.checkpoint
+        dit_config_path = args.config
     config = yaml.safe_load(open(dit_config_path, "r"))
     model_params = recursive_munch(config["model_params"])
     model_params.dit_type = 'DiT'

inference.py

@@ -43,13 +43,13 @@ def load_models(args):
                                                                          "config_dit_mel_seed_uvit_whisper_small_wavenet.yml")
         f0_fn = None
     else:
-        if args.checkpoint_path is None:
+        if args.checkpoint is None:
             dit_checkpoint_path, dit_config_path = load_custom_model_from_hf("Plachta/Seed-VC",
                                                                              "DiT_seed_v2_uvit_whisper_base_f0_44k_bigvgan_pruned_ft_ema_v2.pth",
                                                                              "config_dit_mel_seed_uvit_whisper_base_f0_44k.yml")
         else:
-            dit_checkpoint_path = args.checkpoint_path
-            dit_config_path = args.config_path
+            dit_checkpoint_path = args.checkpoint
+            dit_config_path = args.config
         # f0 extractor
         from modules.rmvpe import RMVPE
 
@@ -414,8 +414,8 @@ def main(args):
     parser.add_argument("--f0-condition", type=str2bool, default=False)
     parser.add_argument("--auto-f0-adjust", type=str2bool, default=False)
     parser.add_argument("--semi-tone-shift", type=int, default=0)
-    parser.add_argument("--checkpoint-path", type=str, help="Path to the checkpoint file", default=None)
-    parser.add_argument("--config-path", type=str, help="Path to the config file", default=None)
+    parser.add_argument("--checkpoint", type=str, help="Path to the checkpoint file", default=None)
+    parser.add_argument("--config", type=str, help="Path to the config file", default=None)
     parser.add_argument("--fp16", type=str2bool, default=True)
     args = parser.parse_args()
     main(args)