Spaces:
Sleeping
Sleeping
Ahmad Bilal
commited on
coreml : add support of large-v1 model (#926)
Browse files
models/convert-whisper-to-coreml.py
CHANGED
|
@@ -296,13 +296,13 @@ def convert_decoder(hparams, model, quantize=False):
|
|
| 296 |
|
| 297 |
if __name__ == "__main__":
|
| 298 |
parser = argparse.ArgumentParser()
|
| 299 |
-
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)", required=True)
|
| 300 |
parser.add_argument("--encoder-only", type=bool, help="only convert encoder", default=False)
|
| 301 |
parser.add_argument("--quantize", type=bool, help="quantize weights to F16", default=False)
|
| 302 |
parser.add_argument("--optimize-ane", type=bool, help="optimize for ANE execution (currently broken)", default=False)
|
| 303 |
args = parser.parse_args()
|
| 304 |
|
| 305 |
-
if args.model not in ["tiny", "tiny.en", "base", "base.en", "small", "small.en", "medium", "medium.en", "large"]:
|
| 306 |
raise ValueError("Invalid model name")
|
| 307 |
|
| 308 |
whisper = load_model(args.model).cpu()
|
|
|
|
| 296 |
|
| 297 |
if __name__ == "__main__":
|
| 298 |
parser = argparse.ArgumentParser()
|
| 299 |
+
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, large-v1)", required=True)
|
| 300 |
parser.add_argument("--encoder-only", type=bool, help="only convert encoder", default=False)
|
| 301 |
parser.add_argument("--quantize", type=bool, help="quantize weights to F16", default=False)
|
| 302 |
parser.add_argument("--optimize-ane", type=bool, help="optimize for ANE execution (currently broken)", default=False)
|
| 303 |
args = parser.parse_args()
|
| 304 |
|
| 305 |
+
if args.model not in ["tiny", "tiny.en", "base", "base.en", "small", "small.en", "medium", "medium.en", "large", "large-v1"]:
|
| 306 |
raise ValueError("Invalid model name")
|
| 307 |
|
| 308 |
whisper = load_model(args.model).cpu()
|