Skip to content

Commit 1ad3471

Browse files
committed
fix a bug in cmd arguments about custom checkpoint and config
1 parent 498d808 commit 1ad3471

File tree

6 files changed

+15
-15
lines changed

6 files changed

+15
-15
lines changed

README-JA.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ python app.py
9191

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

README-ZH.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ python app.py
8181

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

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ This will only load pretrained models for zero-shot inference. To use custom che
9494

9595
Real-time voice conversion GUI:
9696
```bash
97-
python real-time-gui.py --checkpoint <path-to-checkpoint> --config <path-to-config>
97+
python real-time-gui.py --checkpoint-path <path-to-checkpoint> --config-path <path-to-config>
9898
```
9999
- `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`)
100100
- `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

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,14 +20,14 @@ def load_models(args):
2020
print(f"Using device: {device}")
2121
print(f"Using fp16: {fp16}")
2222
# f0 conditioned model
23-
if args.checkpoint_path is None or args.checkpoint_path == "":
23+
if args.checkpoint is None or args.checkpoint == "":
2424
dit_checkpoint_path, dit_config_path = load_custom_model_from_hf("Plachta/Seed-VC",
2525
"DiT_seed_v2_uvit_whisper_base_f0_44k_bigvgan_pruned_ft_ema_v2.pth",
2626
"config_dit_mel_seed_uvit_whisper_base_f0_44k.yml")
2727
else:
28-
print(f"Using custom checkpoint: {args.checkpoint_path}")
29-
dit_checkpoint_path = args.checkpoint_path
30-
dit_config_path = args.config_path
28+
print(f"Using custom checkpoint: {args.checkpoint}")
29+
dit_checkpoint_path = args.checkpoint
30+
dit_config_path = args.config
3131
config = yaml.safe_load(open(dit_config_path, "r"))
3232
model_params = recursive_munch(config["model_params"])
3333
model_params.dit_type = 'DiT'

app_vc.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,13 +19,13 @@ def load_models(args):
1919
fp16 = args.fp16
2020
print(f"Using device: {device}")
2121
print(f"Using fp16: {fp16}")
22-
if args.checkpoint_path is None or args.checkpoint_path == "":
22+
if args.checkpoint is None or args.checkpoint == "":
2323
dit_checkpoint_path, dit_config_path = load_custom_model_from_hf("Plachta/Seed-VC",
2424
"DiT_seed_v2_uvit_whisper_small_wavenet_bigvgan_pruned.pth",
2525
"config_dit_mel_seed_uvit_whisper_small_wavenet.yml")
2626
else:
27-
dit_checkpoint_path = args.checkpoint_path
28-
dit_config_path = args.config_path
27+
dit_checkpoint_path = args.checkpoint
28+
dit_config_path = args.config
2929
config = yaml.safe_load(open(dit_config_path, "r"))
3030
model_params = recursive_munch(config["model_params"])
3131
model_params.dit_type = 'DiT'

inference.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -43,13 +43,13 @@ def load_models(args):
4343
"config_dit_mel_seed_uvit_whisper_small_wavenet.yml")
4444
f0_fn = None
4545
else:
46-
if args.checkpoint_path is None:
46+
if args.checkpoint is None:
4747
dit_checkpoint_path, dit_config_path = load_custom_model_from_hf("Plachta/Seed-VC",
4848
"DiT_seed_v2_uvit_whisper_base_f0_44k_bigvgan_pruned_ft_ema_v2.pth",
4949
"config_dit_mel_seed_uvit_whisper_base_f0_44k.yml")
5050
else:
51-
dit_checkpoint_path = args.checkpoint_path
52-
dit_config_path = args.config_path
51+
dit_checkpoint_path = args.checkpoint
52+
dit_config_path = args.config
5353
# f0 extractor
5454
from modules.rmvpe import RMVPE
5555

@@ -414,8 +414,8 @@ def main(args):
414414
parser.add_argument("--f0-condition", type=str2bool, default=False)
415415
parser.add_argument("--auto-f0-adjust", type=str2bool, default=False)
416416
parser.add_argument("--semi-tone-shift", type=int, default=0)
417-
parser.add_argument("--checkpoint-path", type=str, help="Path to the checkpoint file", default=None)
418-
parser.add_argument("--config-path", type=str, help="Path to the config file", default=None)
417+
parser.add_argument("--checkpoint", type=str, help="Path to the checkpoint file", default=None)
418+
parser.add_argument("--config", type=str, help="Path to the config file", default=None)
419419
parser.add_argument("--fp16", type=str2bool, default=True)
420420
args = parser.parse_args()
421421
main(args)

0 commit comments

Comments
 (0)