mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-13 20:12:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor fixes

Browse Source
This commit is contained in:
zr_jin 2024-11-30 20:12:07 +08:00
parent a1ade8ecb7
commit 6ae48845ba
2 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
egs/wenetspeech4tts/TTS/local/compute_neural_codec_and_prepare_text_tokens.py
View File

@ -516,9 +516,15 @@ def main():
for idx, part in enumerate(cut_sets):
if args.audio_extractor:
if args.audio_extractor == "Encodec":
storage_path = f"{args.output_dir}/{args.prefix}_encodec_{partition}_{idx if split > 1 else ''}"
if split > 1:
storage_path = f"{args.output_dir}/{args.prefix}_encodec_{partition}_{idx}"
else:
storage_path = f"{args.output_dir}/{args.prefix}_encodec_{partition}"
else:
storage_path = f"{args.output_dir}/{args.prefix}_fbank_{partition}_{idx if split > 1 else ''}"
if split > 1:
storage_path = f"{args.output_dir}/{args.prefix}_fbank_{partition}_{idx}"
else:
storage_path = f"{args.output_dir}/{args.prefix}_fbank_{partition}"
if args.prefix.lower() in [
"ljspeech",
@ -587,9 +593,11 @@ def main():
].normalized_text, "normalized_text is None"
# Save each part with an index if split > 1
cuts_filename = (
f"{prefix}cuts_{partition}.{idx if split > 1 else ''}.{args.suffix}"
)
if split > 1:
cuts_filename = f"{prefix}cuts_{partition}.{idx}.{args.suffix}"
else:
cuts_filename = f"{prefix}cuts_{partition}.{args.suffix}"
part.to_file(f"{args.output_dir}/{cuts_filename}")
logging.info(f"Saved {cuts_filename}")

2
egs/wenetspeech4tts/TTS/valle/requirements.txt Normal file
View File

@ -0,0 +1,2 @@
phonemizer==3.2.1
git+https://github.com/facebookresearch/encodec.git