Spaces:
Running
Running
Hưng
commited on
Commit
·
1b2848e
1
Parent(s):
2401617
use ckpt instead of tensors
Browse files- app.py +2 -1
- 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/
|
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
|