speechbrain
/

hifigan-wav2vec2-k1000-LibriTTS

speech-synthesis

Model card Files Files and versions Community

poonehmousavi commited on Jan 10

Commit

06fe4bb

·

verified ·

1 Parent(s): f0dcde3

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -48,7 +48,7 @@ bpe_tokenizers=[None, None]
 vocoder_repo_id = "speechbrain/hifigan-wav2vec2-k1000-LibriTTS"
 kmeans_dataset = "LibriSpeech"
 num_clusters = 1000
-ssl_model = WavLM(model_hub, save_path,output_all_hiddens=True)
 model = DiscreteSSL(save_path, ssl_model, vocoder_repo_id=vocoder_repo_id, kmeans_dataset=kmeans_dataset,num_clusters=num_clusters)
 tokens, _, _ = model.encode(inputs,SSL_layers=ssl_layer_num, deduplicates=deduplicate, bpe_tokenizers=bpe_tokenizers)
 sig = model.decode(tokens, ssl_layer_num)

 vocoder_repo_id = "speechbrain/hifigan-wav2vec2-k1000-LibriTTS"
 kmeans_dataset = "LibriSpeech"
 num_clusters = 1000
+ssl_model = Wav2Vec2(model_hub, save_path,output_all_hiddens=True)
 model = DiscreteSSL(save_path, ssl_model, vocoder_repo_id=vocoder_repo_id, kmeans_dataset=kmeans_dataset,num_clusters=num_clusters)
 tokens, _, _ = model.encode(inputs,SSL_layers=ssl_layer_num, deduplicates=deduplicate, bpe_tokenizers=bpe_tokenizers)
 sig = model.decode(tokens, ssl_layer_num)