We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 648ff93 commit 17b2aa8Copy full SHA for 17b2aa8
examples/asr/transcribe_speech_parallel.py
@@ -215,7 +215,7 @@ def main(cfg: ParallelTranscriptionConfig):
215
logging.info(f"Prediction files are being aggregated in {output_file}.")
216
with open(output_file, 'w') as outf:
217
for rank in range(trainer.world_size):
218
- input_file = os.path.join(cfg.output_path, f"predictions_{rank}.json")
+ input_file = os.path.join(cfg.output_path, f"{cfg.predict_ds.prefix}_predictions_{rank}.json")
219
with open(input_file, 'r') as inpf:
220
lines = inpf.readlines()
221
for line in lines:
0 commit comments