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
|
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):
|