From 42f2747c3828e5dae97c2ba19010c99bb153b860 Mon Sep 17 00:00:00 2001
From: SortAnon <56806675+SortAnon@users.noreply.github.com>
Date: Sat, 24 Jul 2021 21:26:32 +0200
Subject: [PATCH] Bug fix

---
 controllable_talknet.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/controllable_talknet.py b/controllable_talknet.py
index 062f026..678a2ec 100644
--- a/controllable_talknet.py
+++ b/controllable_talknet.py
@@ -879,11 +879,11 @@ def generate_audio(
 
                 spect = tnmodel.force_spectrogram(
                     tokens=tokens,
-                    durs=torch.from_numpy(durs).view(1, -1).to("cuda:0"),
-                    f0=torch.FloatTensor(f0s)
+                    durs=torch.from_numpy(durs)
                     .view(1, -1)
                     .type(torch.LongTensor)
                     .to("cuda:0"),
+                    f0=torch.FloatTensor(f0s).view(1, -1).to("cuda:0"),
                 )
 
             if hifipath != hifigan_path: