Skip to content

Commit 178ec40

Browse files
committed
make preprocessor feature size fit model name
1 parent a5b6391 commit 178ec40

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

.ci/scripts/export_model_cuda_artifact.sh

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,8 +67,12 @@ case "$HF_MODEL" in
6767
TASK="automatic-speech-recognition"
6868
MAX_SEQ_LEN=""
6969
EXTRA_PIP="librosa"
70-
PREPROCESSOR_FEATURE_SIZE="80"
7170
PREPROCESSOR_OUTPUT="whisper_preprocessor.pte"
71+
if [[ "$HF_MODEL" == *"large-v3"* ]]; then
72+
PREPROCESSOR_FEATURE_SIZE="128"
73+
else
74+
PREPROCESSOR_FEATURE_SIZE="80"
75+
fi
7276
;;
7377
google/gemma-3-4b-it)
7478
MODEL_NAME="gemma3"

0 commit comments

Comments
 (0)