NTT123 commited on
Commit
a28cfa9
1 Parent(s): d500e95

update models.

Browse files
app.py CHANGED
@@ -15,11 +15,11 @@ from wavegru_cpp import extract_weight_mask, load_wavegru_cpp
15
 
16
  def speak(text):
17
  alphabet, tacotron_net, tacotron_config = load_tacotron_model(
18
- "./alphabet.txt", "./tacotron.toml", "./tacotrons_ljs_24k_v1_0250000.ckpt"
19
  )
20
 
21
  wavegru_config, wavegru_net = load_wavegru_net(
22
- "./wavegru.yaml", "./wavegru_vocoder_1024_v3_1400000.ckpt"
23
  )
24
 
25
  wave_cpp_weight_mask = extract_weight_mask(wavegru_net)
 
15
 
16
  def speak(text):
17
  alphabet, tacotron_net, tacotron_config = load_tacotron_model(
18
+ "./alphabet.txt", "./tacotron.toml", "./tacotrons_ljs_24k_v1_0300000.ckpt"
19
  )
20
 
21
  wavegru_config, wavegru_net = load_wavegru_net(
22
+ "./wavegru.yaml", "./wavegru_vocoder_1024_v3_1810000.ckpt"
23
  )
24
 
25
  wave_cpp_weight_mask = extract_weight_mask(wavegru_net)
extract_model.py CHANGED
@@ -1,5 +1,5 @@
1
  import pickle
2
 
3
- dic = pickle.load(open("./wavegru_vocoder_1024_v3_1400000.ckpt", "rb"))
4
  del dic["optim_state_dict"]
5
- pickle.dump(dic, open("./wavegru_vocoder_1024_v3_1400000.ckpt", "wb"))
 
1
  import pickle
2
 
3
+ dic = pickle.load(open("./tacotrons_ljs_24k_v1_0300000.ckpt", "rb"))
4
  del dic["optim_state_dict"]
5
+ pickle.dump(dic, open("./tacotrons_ljs_24k_v1_0300000.ckpt", "wb"))
tacotrons_ljs_24k_v1_0250000.ckpt → tacotrons_ljs_24k_v1_0300000.ckpt RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:512b3af6ef95ccc53d3516256abae81b025e110fa886ec68f9f7033039013fc6
3
  size 53561547
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:73368d52c6c519682b73fa9676fe2eaed1712aa559026034ab6a36b2bfd8f8c0
3
  size 53561547
wavegru_vocoder_1024_v3_1400000.ckpt → wavegru_vocoder_1024_v3_1810000.ckpt RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9e5a9a2e67872e8a0fa89a34f898be41ef51838dc534cd281aded3c39a9a1017
3
  size 69717674
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e0795c5b67b49c79e801f3934e8e443ba5237baf2c8361085c29aaee9da43a00
3
  size 69717674