added version parameter
This commit is contained in:
@@ -21,6 +21,11 @@ async def main() -> None:
|
|||||||
default="medium",
|
default="medium",
|
||||||
help="Name of whisper model to use (default medium)",
|
help="Name of whisper model to use (default medium)",
|
||||||
)
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"--language",
|
||||||
|
default="en",
|
||||||
|
help="Language to use for inference (default: en)",
|
||||||
|
)
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--uri", help="unix:// or tcp://", default="tcp://0.0.0.0:10300"
|
"--uri", help="unix:// or tcp://", default="tcp://0.0.0.0:10300"
|
||||||
)
|
)
|
||||||
@@ -34,16 +39,17 @@ async def main() -> None:
|
|||||||
default="cuda",
|
default="cuda",
|
||||||
help="Device to use for inference (default: cuda)",
|
help="Device to use for inference (default: cuda)",
|
||||||
)
|
)
|
||||||
parser.add_argument(
|
|
||||||
"--language",
|
|
||||||
default="en",
|
|
||||||
help="Default language to set for transcription (default: en)",
|
|
||||||
)
|
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--beam-size",
|
"--beam-size",
|
||||||
type=int,
|
type=int,
|
||||||
default=5,
|
default=5,
|
||||||
)
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"--version",
|
||||||
|
default="1.0",
|
||||||
|
help="Version of the whisper model to use (default: 1.0)",
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
parser.add_argument("--debug", action="store_true", help="Log DEBUG messages")
|
parser.add_argument("--debug", action="store_true", help="Log DEBUG messages")
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
@@ -61,6 +67,7 @@ async def main() -> None:
|
|||||||
AsrProgram(
|
AsrProgram(
|
||||||
name="whisper",
|
name="whisper",
|
||||||
description="Whisper",
|
description="Whisper",
|
||||||
|
version=args.version,
|
||||||
attribution=Attribution(
|
attribution=Attribution(
|
||||||
name="OpenAI",
|
name="OpenAI",
|
||||||
url="https://github.com/openai/whisper",
|
url="https://github.com/openai/whisper",
|
||||||
@@ -69,6 +76,7 @@ async def main() -> None:
|
|||||||
models=[
|
models=[
|
||||||
AsrModel(
|
AsrModel(
|
||||||
name=args.model,
|
name=args.model,
|
||||||
|
version=args.version,
|
||||||
description=args.model,
|
description=args.model,
|
||||||
attribution=Attribution(
|
attribution=Attribution(
|
||||||
name="whisper",
|
name="whisper",
|
||||||
|
|||||||
Reference in New Issue
Block a user