File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff 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'
You can’t perform that action at this time.
0 commit comments