mirror of
https://github.com/storytold/artcraft.git
synced 2026-11-06 02:00:46 +00:00
this works
This commit is contained in:
1 parent
91dea21a2f
commit
6fdbaa2ad1
3 files changed
+4
-4
No files matched your search
@@ -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";
|
||||
|
||||
@@ -601,4 +601,4 @@ class Tacotron2(nn.Module):
|
||||
])
|
||||
|
||||
#return outputs
|
||||
return outputs[0]
|
||||
return outputs[0].float()
|
||||
@@ -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
|
||||
|
||||
Reference in new issue
Block a user