Respair commited on
Commit
f77b559
·
verified ·
1 Parent(s): 19b3a06

Update pkanade_24_multi_gpu_train_finetune_accelerate.py

Browse files
pkanade_24_multi_gpu_train_finetune_accelerate.py CHANGED
@@ -712,7 +712,7 @@ def main(config_path):
712
  gt = torch.stack(gt).detach()
713
  s = model.predictor_encoder(gt.unsqueeze(0))
714
 
715
- F0_fake, N_fake = model.predictor.F0Ntrain(p_en, s)
716
 
717
  loss_dur = 0
718
  for _s2s_pred, _text_input, _text_length in zip(d, (d_gt), input_lengths):
 
712
  gt = torch.stack(gt).detach()
713
  s = model.predictor_encoder(gt.unsqueeze(0))
714
 
715
+ F0_fake, N_fake = model.predictor(texts=p_en, style=s_dur, f0=True)
716
 
717
  loss_dur = 0
718
  for _s2s_pred, _text_input, _text_length in zip(d, (d_gt), input_lengths):