ExternalVendorCode/whisper.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2025-06-20 15:40:28 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

whisper : add large-v3-turbo (#2440)

Browse Source
This commit is contained in:
Georgi Gerganov
2024-10-01 15:57:06 +03:00
committed by GitHub
parent 8feb375fbd
commit 2ef717b293
16 changed files with 39 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
models/convert-whisper-to-openvino.py
View File

@ -45,10 +45,10 @@ def convert_encoder(hparams, encoder, mname):
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--model", type=str, help="model to convert (e.g. tiny, tiny.en, base, base.en, small, small.en, medium, medium.en, large-v1, large-v2, large-v3)", required=True)
parser.add_argument("--model", type=str, help="model to convert (e.g. tiny, tiny.en, base, base.en, small, small.en, medium, medium.en, large-v1, large-v2, large-v3, large-v3-turbo)", required=True)
args = parser.parse_args()
if args.model not in ["tiny", "tiny.en", "base", "base.en", "small", "small.en", "medium", "medium.en", "large-v1", "large-v2", "large-v3"]:
if args.model not in ["tiny", "tiny.en", "base", "base.en", "small", "small.en", "medium", "medium.en", "large-v1", "large-v2", "large-v3", "large-v3-turbo"]:
raise ValueError("Invalid model name")
whisper = load_model(args.model).cpu()