We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a5b6391 commit 178ec40Copy full SHA for 178ec40
.ci/scripts/export_model_cuda_artifact.sh
@@ -67,8 +67,12 @@ case "$HF_MODEL" in
67
TASK="automatic-speech-recognition"
68
MAX_SEQ_LEN=""
69
EXTRA_PIP="librosa"
70
- PREPROCESSOR_FEATURE_SIZE="80"
71
PREPROCESSOR_OUTPUT="whisper_preprocessor.pte"
+ if [[ "$HF_MODEL" == *"large-v3"* ]]; then
72
+ PREPROCESSOR_FEATURE_SIZE="128"
73
+ else
74
+ PREPROCESSOR_FEATURE_SIZE="80"
75
+ fi
76
;;
77
google/gemma-3-4b-it)
78
MODEL_NAME="gemma3"
0 commit comments