Skip to content

Commit 2ba39ce

Browse files
committed
[aoti-et] Add Whisper large
1 parent 357d573 commit 2ba39ce

File tree

3 files changed

+19
-1
lines changed

3 files changed

+19
-1
lines changed

.ci/scripts/test_model_cuda_e2e.sh

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ Arguments:
1818
Supported models:
1919
- mistralai/Voxtral-Mini-3B-2507
2020
- openai/whisper-small
21+
- openai/whisper-large
2122
- google/gemma-3-4b-it
2223
2324
quant_name Quantization type (required)
@@ -103,6 +104,18 @@ case "$HF_MODEL" in
103104
AUDIO_FILE="output.wav"
104105
IMAGE_PATH=""
105106
;;
107+
openai/whisper-large)
108+
MODEL_NAME="whisper"
109+
RUNNER_TARGET="whisper_runner"
110+
RUNNER_PATH="whisper"
111+
EXPECTED_OUTPUT="Mr. Quilter is the apostle of the middle classes"
112+
PREPROCESSOR="whisper_preprocessor.pte"
113+
TOKENIZER_URL="https://huggingface.co/openai/whisper-large/resolve/main" # @lint-ignore
114+
TOKENIZER_FILE=""
115+
AUDIO_URL=""
116+
AUDIO_FILE="output.wav"
117+
IMAGE_PATH=""
118+
;;
106119
google/gemma-3-4b-it)
107120
MODEL_NAME="gemma3"
108121
RUNNER_TARGET="gemma3_e2e_runner"
@@ -117,7 +130,7 @@ case "$HF_MODEL" in
117130
;;
118131
*)
119132
echo "Error: Unsupported model '$HF_MODEL'"
120-
echo "Supported models: mistralai/Voxtral-Mini-3B-2507, openai/whisper-small, google/gemma-3-4b-it"
133+
echo "Supported models: mistralai/Voxtral-Mini-3B-2507, openai/whisper-small, openai/whisper-large, google/gemma-3-4b-it"
121134
exit 1
122135
;;
123136
esac

.github/workflows/cuda.yml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,6 +104,8 @@ jobs:
104104
name: "Voxtral-Mini-3B-2507"
105105
- repo: "openai"
106106
name: "whisper-small"
107+
- repo: "openai"
108+
name: "whisper-large"
107109
- repo: "google"
108110
name: "gemma-3-4b-it"
109111
quant:
@@ -223,6 +225,8 @@ jobs:
223225
name: "Voxtral-Mini-3B-2507"
224226
- repo: "openai"
225227
name: "whisper-small"
228+
- repo: "openai"
229+
name: "whisper-large"
226230
- repo: "google"
227231
name: "gemma-3-4b-it"
228232
quant:

runtime/core/exec_aten/exec_aten.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@
3737
#include <executorch/runtime/core/portable_type/complex.h> // @manual
3838
#include <executorch/runtime/core/portable_type/device.h> // @manual
3939
#include <executorch/runtime/core/portable_type/half.h> // @manual
40+
#include <executorch/runtime/core/portable_type/optional.h> // @manual
4041
#include <executorch/runtime/core/portable_type/qint_types.h> // @manual
4142
#include <executorch/runtime/core/portable_type/scalar.h> // @manual
4243
#include <executorch/runtime/core/portable_type/scalar_type.h> // @manual

0 commit comments

Comments
 (0)