Skip to content

Commit 4b0100f

Browse files
committed
更新训练与微调脚本,更新tiny版本模型
1 parent 00750bf commit 4b0100f

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

eval.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -55,11 +55,9 @@ def calc_mos(computor, audio, orin_sr):
5555
)
5656

5757
def load_models(args):
58-
# dit_checkpoint_path, dit_config_path = load_custom_model_from_hf("Plachta/Seed-VC",
59-
# "DiT_seed_v2_uvit_whisper_small_wavenet_bigvgan_pruned.pth",
60-
# "config_dit_mel_seed_uvit_whisper_small_wavenet.yml")
61-
dit_checkpoint_path = "../DiT_epoch_00012_step_481500_xlsr_ema.pth"
62-
dit_config_path = "./configs/config_dit_mel_seed_uvit_xlsr_tiny.yml"
58+
dit_checkpoint_path, dit_config_path = load_custom_model_from_hf("Plachta/Seed-VC",
59+
"DiT_seed_v2_uvit_whisper_small_wavenet_bigvgan_pruned.pth",
60+
"config_dit_mel_seed_uvit_whisper_small_wavenet.yml")
6361
config = yaml.safe_load(open(dit_config_path, "r"))
6462
model_params = recursive_munch(config["model_params"])
6563
model = build_model(model_params, stage="DiT")

0 commit comments

Comments
 (0)