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
|