mirror of https://github.com/coqui-ai/TTS.git
Fix for 'NoneType' object has no attribute 'to'
This commit is contained in:
parent
28a64221ea
commit
2aa0354b44
|
@ -37,6 +37,9 @@ def tts(model, vocoder_model, text, CONFIG, use_cuda, ap, use_gl, speaker_fileid
|
|||
if CONFIG.model == "Tacotron" and not use_gl:
|
||||
mel_postnet_spec = ap.out_linear_to_mel(mel_postnet_spec.T).T
|
||||
if not use_gl:
|
||||
beta = np.linspace(1e-6, 0.01, 50)
|
||||
vocoder_model.compute_noise_level(beta)
|
||||
|
||||
device_type = "cuda" if use_cuda else "cpu"
|
||||
waveform = vocoder_model.inference(torch.FloatTensor(mel_postnet_spec.T).to(device_type).unsqueeze(0))
|
||||
if use_cuda and not use_gl:
|
||||
|
|
Loading…
Reference in New Issue