15 lines
861 B
Python
15 lines
861 B
Python
|
import argparse
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument("--conv-frontend", type=str, required=True)
|
||
|
|
parser.add_argument("--encoder", type=str, required=True)
|
||
|
|
parser.add_argument("--decoder", type=str, required=True)
|
||
|
|
parser.add_argument("--tokenizer", type=str, required=True)
|
||
|
|
parser.add_argument("--language", type=str, default="", help="Force language, e.g. Indonesian, English, Chinese")
|
||
|
|
parser.add_argument("--hotwords", type=str, default="", help="Comma-separated hotword phrases, e.g. 'foo,bar'")
|
||
|
|
parser.add_argument("--num-threads", type=int, default=2)
|
||
|
|
parser.add_argument("--provider", type=str, default="cpu", choices=["cpu", "cuda"])
|
||
|
|
parser.add_argument("--max-total-len", type=int, default=2048)
|
||
|
|
parser.add_argument("--max-new-tokens", type=int, default=256)
|
||
|
|
parser.add_argument("sounds", nargs="+", help="Audio files to transcribe")
|