this works

This commit is contained in:
Brandon Thomas committed 2020-01-05 03:29:17 -05:00
1 parent 91dea21a2f
commit 6fdbaa2ad1
3 files changed
+4 -4

No files matched your search

+1 -1
View File
@@ -97,7 +97,7 @@ pub fn run_server() {
.mount("/pub", StaticFiles::new("/www/public", options).rank(-1))*/
//let tacotron = "/home/bt/dev/voder/tacotron_melgan/tacotron_jit_model_voder_c0cac635.pt";
let tacotron = "/home/bt/dev/voder/tacotron_melgan/tacotron_container2.pt";
let tacotron = "/home/bt/dev/voder/tacotron_melgan/tacotron_container3.pt";
//let melgan = "/home/bt/dev/voder/tacotron_melgan/melgan_jit_model_voder_c0cac635.pt";
let melgan = "/home/bt/dev/voder/tacotron_melgan/melgan_container2.pt";
let melgan = "/home/bt/dev/tacotron-melgan/shared_melgan_container.pt";
+1 -1
View File
@@ -601,4 +601,4 @@ class Tacotron2(nn.Module):
])
#return outputs
return outputs[0]
return outputs[0].float()
+2 -2
View File
@@ -11,7 +11,7 @@ from math import sqrt
from tacotron_model import Tacotron2
torch.set_default_tensor_type('torch.DoubleTensor')
#torch.set_default_tensor_type('torch.DoubleTensor')
#torch.set_default_tensor_type('torch.FloatTensor')
print('Loading tacotron model...')
@@ -77,7 +77,7 @@ print('Load state dict...')
module.load_state_dict(tacotron_model['state_dict'])
#module.eval() # NB: Complains unless called: Did you forget call .eval() on your model?
output_filename = 'tacotron_container2.pt'
output_filename = 'tacotron_container3.pt'
# NB: Tracing evaluates the model on input and unrolls and hardcodes branching
# and loops. Scripting allows these to remain by converting the entire program