daniel543x/artcraft
1
0 Fork 0
mirror of https://github.com/storytold/artcraft.git synced 2026-11-05 09:48:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

rename models

Browse Source
This commit is contained in:
Brandon Thomas committed 2020-01-02 01:19:49 -05:00
1 parent c0cac635f4
commit 1ca7ee8a5b
5 files changed
+18 -8
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

tacotron_melgan/src/main.rs
+2 -2
View File
@@ -33,11 +33,11 @@ pub fn main() {
fn run_end_to_end() {
println!("Load Tacotron Model");
let tacotron_model =
model::load_model_file("/home/bt/dev/voder/tacotron_melgan/tacotron_container.pt");
model::load_model_file("/home/bt/dev/voder/tacotron_melgan/tacotron_jit_model_voder_c0cac635.pt");
println!("Load Melgan Model");
let melgan_model =
model::load_model_file("/home/bt/dev/voder/tacotron_melgan/container2.pt");
model::load_model_file("/home/bt/dev/voder/tacotron_melgan/melgan_jit_model_voder_c0cac635.pt");
println!("Load Text Sequence Tensor");
let text_sequence =
tacotron_melgan/src/melgan.rs
+1 -1
View File
@@ -10,7 +10,7 @@ use model::{
load_wrapped_tensor_file,
};
const WRAPPED_MODEL_PATH : &'static str = "/home/bt/dev/voder/tacotron_melgan/container2.pt";
const WRAPPED_MODEL_PATH : &'static str = "/home/bt/dev/voder/tacotron_melgan/melgan_jit_model_voder_c0cac635.pt";
const EXAMPLE_MEL_1: &'static str = "/home/bt/dev/voder/data/mels/LJ002-0320.mel.containerized.pt";
const EXAMPLE_MEL_2 : &'static str = "/home/bt/dev/voder/data/mels/trump_2018_02_15-001.mel.containerized.pt";
tacotron_melgan/src/model.rs
+12 -2
View File
@@ -47,10 +47,10 @@ impl TacoMelModel {
pub fn create(tacotron_filename: &str, melgan_filename: &str) -> Self {
let tacotron_model =
load_model_file("/home/bt/dev/voder/tacotron_melgan/tacotron_container.pt");
load_model_file("/home/bt/dev/voder/tacotron_melgan/tacotron_jit_model_voder_c0cac635.pt");
let melgan_model =
load_model_file("/home/bt/dev/voder/tacotron_melgan/container2.pt");
load_model_file("/home/bt/dev/voder/tacotron_melgan/melgan_jit_model_voder_c0cac635.pt");
Self {
tacotron_model,
@@ -80,6 +80,16 @@ impl TacoMelModel {
text_buffer.push(v);
}
/*let mut text_buffer4: Vec<i64> = Vec::new();
text_buffer.push(45);
text_buffer.push(46);
text_buffer.push(11);
text_buffer.push(62);
text_buffer.push(52);
text_buffer.push(58);*/
println!("Text buffer: {:?}", text_buffer);
let text_tensor = Tensor::of_slice(text_buffer.as_slice());
println!("Text tensor: {:?}", text_tensor);
tacotron_melgan/src/server.rs
+2 -2
View File
@@ -40,8 +40,8 @@ pub fn run_server() {
.mount("/pub", StaticFiles::new("/www/public", options).rank(-1))*/
let model = TacoMelModel::create(
"/home/bt/dev/voder/tacotron_melgan/tacotron_container.pt",
"/home/bt/dev/voder/tacotron_melgan/container2.pt");
"/home/bt/dev/voder/tacotron_melgan/tacotron_jit_model_voder_c0cac635.pt",
"/home/bt/dev/voder/tacotron_melgan/melgan_jit_model_voder_c0cac635.pt");
rocket::ignite()
.mount("/", routes![
tacotron_melgan/src/tacotron.rs
+1 -1
View File
@@ -4,7 +4,7 @@ use tch;
use tch::nn::Module;
const WRAPPED_MODEL_PATH : &'static str = "/home/bt/dev/voder/tacotron_melgan/tacotron_container.pt";
const WRAPPED_MODEL_PATH : &'static str = "/home/bt/dev/voder/tacotron_melgan/tacotron_jit_model_voder_c0cac635.pt";
//const EXAMPLE_TEXT_SEQUENCE : &'static str = "/home/bt/dev/voder/data/text/tacotron_text_sequence.pt";
const EXAMPLE_TEXT_SEQUENCE : &'static str = "/home/bt/dev/voder/data/text/tacotron_text_sequence.pt.containerized.pt";