mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-14 12:32:20 +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): for idx, part in enumerate(cut_sets):
if args.audio_extractor: if args.audio_extractor:
if args.audio_extractor == "Encodec": 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: 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 [ if args.prefix.lower() in [
"ljspeech", "ljspeech",
@ -587,9 +593,11 @@ def main():
].normalized_text, "normalized_text is None" ].normalized_text, "normalized_text is None"
# Save each part with an index if split > 1 # Save each part with an index if split > 1
cuts_filename = ( if split > 1:
f"{prefix}cuts_{partition}.{idx if split > 1 else ''}.{args.suffix}" 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}") part.to_file(f"{args.output_dir}/{cuts_filename}")
logging.info(f"Saved {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