Spaces:
Runtime error
Runtime error
GreenRaptor
commited on
Commit
•
3533765
1
Parent(s):
edf267e
Update mms_infer.py
Browse files- 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,)
|