diff --git a/scripts/sense-voice/export-onnx.py b/scripts/sense-voice/export-onnx.py index 97d9a506..48b68636 100755 --- a/scripts/sense-voice/export-onnx.py +++ b/scripts/sense-voice/export-onnx.py @@ -119,7 +119,7 @@ def display_params(params): def main(): - model, params = SenseVoiceSmall.from_pretrained(model="iic/SenseVoiceSmall") + model, params = SenseVoiceSmall.from_pretrained(model="iic/SenseVoiceSmall", device="cpu") display_params(params) generate_tokens(params)