Yixuan Li commited on
Commit
f0c790f
·
1 Parent(s): 56879f2

revise ckpt paths

Browse files
Files changed (1) hide show
  1. 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: /hpc_stor03/sjtu_home/xuenan.xu/workspace/text_to_audio_generation/ezaudio/ckpts/vae/1m.pt
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: /hpc_stor03/sjtu_home/zeyu.xie/workspace/speech2audio/hear/output/qformer_caption_tts_hubert/exp0_best.pt
84
- pretrained_ckpt: /hpc_stor03/sjtu_home/zeyu.xie/workspace/speech2audio/x2audio/x_to_audio_generation/experiments/audiocaps_fm/checkpoints/epoch_100/model.safetensors
 
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