Skip to content

Commit 17b2aa8

Browse files
Adding prefix option for transcribe to allow multiple parallel runs
Signed-off-by: Nune <[email protected]>
1 parent 648ff93 commit 17b2aa8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/asr/transcribe_speech_parallel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -215,7 +215,7 @@ def main(cfg: ParallelTranscriptionConfig):
215215
logging.info(f"Prediction files are being aggregated in {output_file}.")
216216
with open(output_file, 'w') as outf:
217217
for rank in range(trainer.world_size):
218-
input_file = os.path.join(cfg.output_path, f"predictions_{rank}.json")
218+
input_file = os.path.join(cfg.output_path, f"{cfg.predict_ds.prefix}_predictions_{rank}.json")
219219
with open(input_file, 'r') as inpf:
220220
lines = inpf.readlines()
221221
for line in lines:

0 commit comments

Comments
 (0)