From 825ffa47d18994510882f0fa4f2708f4359d5377 Mon Sep 17 00:00:00 2001
From: rafaelvalle <rafaelvalle@nvidia.com>
Date: Sat, 8 Dec 2018 21:26:01 -0800
Subject: [PATCH] inference.ipynb: reverting fp16 inference for now

---
 inference.ipynb | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/inference.ipynb b/inference.ipynb
index 3e25002..f78d5af 100644
--- a/inference.ipynb
+++ b/inference.ipynb
@@ -116,9 +116,7 @@
    "source": [
     "waveglow_path = 'waveglow_old.pt'\n",
     "waveglow = torch.load(waveglow_path)['model']\n",
-    "waveglow.cuda().half()\n",
-    "for k in waveglow.convinv:\n",
-    "    k.float()"
+    "waveglow.cuda()"
    ]
   },
   {
@@ -205,7 +203,7 @@
    ],
    "source": [
     "with torch.no_grad():\n",
-    "    audio = waveglow.infer(mel_outputs_postnet.half(), sigma=0.666)\n",
+    "    audio = waveglow.infer(mel_outputs_postnet, sigma=0.666)\n",
     "ipd.Audio(audio[0].data.cpu().numpy(), rate=hparams.sampling_rate)"
    ]
   }