Hưng commited on
Commit
1b2848e
·
1 Parent(s): 2401617

use ckpt instead of tensors

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. requirements.txt +0 -1
app.py CHANGED
@@ -23,8 +23,9 @@ model = load_model(
23
  DiT,
24
  dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4),
25
  ckpt_path=str(
26
- cached_path("hf://whatvn/vietnamese-tts/model.tensors")
27
  ),
 
28
  vocab_file=str(cached_path("hf://whatvn/vietnamese-tts/vocab.txt")),
29
  )
30
 
 
23
  DiT,
24
  dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4),
25
  ckpt_path=str(
26
+ cached_path("hf://whatvn/vietnamese-tts/model_960000.pt")
27
  ),
28
+ mel_spec_type="vocos",
29
  vocab_file=str(cached_path("hf://whatvn/vietnamese-tts/vocab.txt")),
30
  )
31
 
requirements.txt CHANGED
@@ -1,4 +1,3 @@
1
- accelerate
2
  torch
3
  torchaudio
4
  soundfile
 
 
1
  torch
2
  torchaudio
3
  soundfile