Spaces:
Runtime error
Runtime error
Yixuan Li
commited on
Commit
·
f0c790f
1
Parent(s):
56879f2
revise ckpt paths
Browse files- configs/config.yaml +4 -3
configs/config.yaml
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
model:
|
| 2 |
autoencoder:
|
| 3 |
_target_: models.autoencoder.waveform.stable_vae.StableVAE
|
|
@@ -38,7 +39,7 @@ model:
|
|
| 38 |
latent_dim: 128
|
| 39 |
downsampling_ratio: 480
|
| 40 |
sample_rate: 24000
|
| 41 |
-
pretrained_ckpt:
|
| 42 |
bottleneck:
|
| 43 |
_target_: models.autoencoder.waveform.stable_vae.VAEBottleneck
|
| 44 |
backbone:
|
|
@@ -80,5 +81,5 @@ model:
|
|
| 80 |
text_encoder: None
|
| 81 |
speech_encoder:
|
| 82 |
_target_: models.content_encoder.star_encoder.star_encoder.QformerBridgeNet
|
| 83 |
-
load_from_pretrained: /
|
| 84 |
-
pretrained_ckpt:
|
|
|
|
| 1 |
+
sample_rate: 24000
|
| 2 |
model:
|
| 3 |
autoencoder:
|
| 4 |
_target_: models.autoencoder.waveform.stable_vae.StableVAE
|
|
|
|
| 39 |
latent_dim: 128
|
| 40 |
downsampling_ratio: 480
|
| 41 |
sample_rate: 24000
|
| 42 |
+
pretrained_ckpt: ckpts/1m.pt
|
| 43 |
bottleneck:
|
| 44 |
_target_: models.autoencoder.waveform.stable_vae.VAEBottleneck
|
| 45 |
backbone:
|
|
|
|
| 81 |
text_encoder: None
|
| 82 |
speech_encoder:
|
| 83 |
_target_: models.content_encoder.star_encoder.star_encoder.QformerBridgeNet
|
| 84 |
+
load_from_pretrained: ckpts/exp0_best.pt
|
| 85 |
+
pretrained_ckpt: model.safetensors
|