更新训练与微调脚本，更新tiny版本模型

Plachtaa · Nov 26, 2024 · 4b0100f · 4b0100f
1 parent 00750bf
commit 4b0100f
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/eval.py b/eval.py
@@ -55,11 +55,9 @@ def calc_mos(computor, audio, orin_sr):
 )
 
 def load_models(args):
-    # 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")
-    dit_checkpoint_path = "../DiT_epoch_00012_step_481500_xlsr_ema.pth"
-    dit_config_path = "./configs/config_dit_mel_seed_uvit_xlsr_tiny.yml"
+    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")
     config = yaml.safe_load(open(dit_config_path, "r"))
     model_params = recursive_munch(config["model_params"])
     model = build_model(model_params, stage="DiT")