diff --git a/inference.ipynb b/inference.ipynb index 83a23f7e2..74e78d739 100644 --- a/inference.ipynb +++ b/inference.ipynb @@ -87,7 +87,7 @@ "checkpoint_path = \"tacotron2_statedict.pt\"\n", "model = load_model(hparams)\n", "model.load_state_dict(torch.load(checkpoint_path)['state_dict'])\n", - "_ = model.cuda().eval().half()" + "_ = model.cuda().eval() #.half()" ] }, { @@ -111,7 +111,7 @@ "source": [ "waveglow_path = 'waveglow_256channels.pt'\n", "waveglow = torch.load(waveglow_path)['model']\n", - "waveglow.cuda().eval().half()\n", + "waveglow.cuda().eval() #.half()\n", "for k in waveglow.convinv:\n", " k.float()\n", "denoiser = Denoiser(waveglow)"