Skip to content

Commit 0b4decd

Browse files
committed
solve comments
1 parent 98a3c7d commit 0b4decd

File tree

4 files changed

+6
-8
lines changed

4 files changed

+6
-8
lines changed

.ci/scripts/export_model_cuda_artifact.sh

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,7 @@ Arguments:
1717
hf_model HuggingFace model ID (required)
1818
Supported models:
1919
- mistralai/Voxtral-Mini-3B-2507
20-
- openai/whisper-small
21-
- openai/whisper-large-v2
20+
- openai/whisper series (whisper-{small, medium, large, large-v2, large-v3, large-v3-turbo})
2221
- google/gemma-3-4b-it
2322
2423
quant_name Quantization type (optional, default: non-quantized)
@@ -81,7 +80,7 @@ case "$HF_MODEL" in
8180
;;
8281
*)
8382
echo "Error: Unsupported model '$HF_MODEL'"
84-
echo "Supported models: mistralai/Voxtral-Mini-3B-2507, openai/whisper-small, openai/whisper-large-v2, google/gemma-3-4b-it"
83+
echo "Supported models: mistralai/Voxtral-Mini-3B-2507, openai/whisper-{small, medium, large, large-v2, large-v3, large-v3-turbo}, google/gemma-3-4b-it"
8584
exit 1
8685
;;
8786
esac

.ci/scripts/test_model_cuda_e2e.sh

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,7 @@ Arguments:
1717
hf_model HuggingFace model ID (required)
1818
Supported models:
1919
- mistralai/Voxtral-Mini-3B-2507
20-
- openai/whisper-small
21-
- openai/whisper-large-v2
20+
- openai/whisper series (whisper-{small, medium, large, large-v2, large-v3, large-v3-turbo})
2221
- google/gemma-3-4b-it
2322
2423
quant_name Quantization type (required)

.github/workflows/cuda.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,7 @@ jobs:
226226
- repo: "openai"
227227
name: "whisper-small"
228228
- repo: "openai"
229-
name: "whisper-large-v2"
229+
name: "whisper-large-v3-turbo"
230230
- repo: "google"
231231
name: "gemma-3-4b-it"
232232
quant:

examples/models/whisper/main.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ DEFINE_string(
4242
DEFINE_string(
4343
model_name,
4444
"base",
45-
"Whisper model name (base, small, medium, large, large-v2, large-v3, turbo).");
45+
"Whisper model name (base, small, medium, large, large-v2, large-v3, large-v3-turbo).");
4646
DEFINE_double(
4747
temperature,
4848
0.0,
@@ -116,7 +116,7 @@ int main(int argc, char** argv) {
116116

117117
// Set decoder_start_token_id based on model version
118118
if (FLAGS_model_name == "large-v2" || FLAGS_model_name == "large-v3" ||
119-
FLAGS_model_name == "turbo") {
119+
FLAGS_model_name == "large-v3-turbo") {
120120
config.decoder_start_token_id = 50258;
121121
ET_LOG(
122122
Info,

0 commit comments

Comments
 (0)