GreenRaptor commited on
Commit
3533765
1 Parent(s): edf267e

Update mms_infer.py

Browse files
Files changed (1) hide show
  1. mms_infer.py +1 -1
mms_infer.py CHANGED
@@ -37,7 +37,7 @@ def process(args):
37
  with open(tmpdir / "dev.wrd", "w") as fw:
38
  fw.write("dummy dummy\n"*len(args.audio))
39
  cmd = f"""
40
- PYTHONPATH=. PREFIX=INFER HYDRA_FULL_ERROR=1 python infer.py -m dataset.max_tokens=4000000 distributed_training.distributed_world_size=1 "common_eval.path='{args.model}'" task.data={tmpdir} dataset.gen_subset="{args.lang}:dev" common_eval.post_process={args.format} decoding.results_path={tmpdir}
41
  """
42
  print(">>> loading model & running inference ...", file=sys.stderr)
43
  subprocess.run(cmd, shell=True, stdout=subprocess.DEVNULL,)
 
37
  with open(tmpdir / "dev.wrd", "w") as fw:
38
  fw.write("dummy dummy\n"*len(args.audio))
39
  cmd = f"""
40
+ PYTHONPATH=. PREFIX=INFER HYDRA_FULL_ERROR=1 python infer.py -m decoding.type=viterbi dataset.max_tokens=4000000 distributed_training.distributed_world_size=1 "common_eval.path='{args.model}'" task.data={tmpdir} dataset.gen_subset="{args.lang}:dev" common_eval.post_process={args.format} decoding.results_path={tmpdir}
41
  """
42
  print(">>> loading model & running inference ...", file=sys.stderr)
43
  subprocess.run(cmd, shell=True, stdout=subprocess.DEVNULL,)