Support distil-whisper (#411)

scripts/whisper/export-onnx.py

@@ -39,7 +39,9 @@ def get_args():
         choices=[
             "tiny", "tiny.en", "base", "base.en",
             "small", "small.en", "medium", "medium.en",
-            "large", "large-v1", "large-v2"],
+            "large", "large-v1", "large-v2",
+            "distil-medium.en",
+            ],
         # fmt: on
     )
     return parser.parse_args()
@@ -257,10 +259,27 @@ def convert_tokens(name, model):
 def main():
     args = get_args()
     name = args.model
+    print(args)
+    print(name)
 
     opset_version = 13
 
-    model = whisper.load_model(name)
+    if name == "distil-medium.en":
+        filename = "./distil-medium-en-original-model.bin"
+        if not Path(filename):
+            raise ValueError(
+                """
+                Please go to https://huggingface.co/distil-whisper/distil-medium.en
+                to download original-model.bin
+                You can use the following command to do that:
+
+                wget -O distil-medium-en-original-model.bin https://huggingface.co/distil-whisper/distil-medium.en/resolve/main/original-model.bin
+            """
+            )
+        model = whisper.load_model(filename)
+    else:
+        model = whisper.load_model(name)
+
     print(
         f"number of model parameters: {name}",
         sum(p.numel() for p in model.parameters()),