Skip to content

Commit

Permalink
small fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
csukuangfj committed Aug 15, 2023
1 parent aa076de commit 6cc87c0
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions scripts/whisper/export-onnx.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@
"""

import argparse
import os
from pathlib import Path
from typing import Any, Dict, Optional

Expand Down Expand Up @@ -418,6 +419,9 @@ def main():
},
)

if 'large' in args.model:
# it causes errors for large models, so skip it.
return
# Generate int8 quantization models
# See https://onnxruntime.ai/docs/performance/model-optimizations/quantization.html#data-type-selection

Expand Down

0 comments on commit 6cc87c0

Please sign in to comment.