11#! /bin/bash
2- # (c) Meta Platforms, Inc. and affiliates. Confidential and proprietary.
2+ # Copyright (c) Meta Platforms, Inc. and affiliates.
3+ # All rights reserved.
4+ #
5+ # This source code is licensed under the BSD-style license found in the
6+ # LICENSE file in the root directory of this source tree.
37
4- # Test CUDA model end-to-end
8+ # Test CUDA model end-to-end, need to run .ci/scripts/export_model_cuda_artifact.sh first
59
610show_help () {
711 cat << EOF
@@ -81,7 +85,7 @@ case "$HF_MODEL" in
8185 RUNNER_PATH=" voxtral"
8286 EXPECTED_OUTPUT=" poem"
8387 PREPROCESSOR=" voxtral_preprocessor.pte"
84- TOKENIZER_URL=" https://huggingface.co/mistralai/Voxtral-Mini-3B-2507/resolve/main"
88+ TOKENIZER_URL=" https://huggingface.co/mistralai/Voxtral-Mini-3B-2507/resolve/main" # @lint-ignore
8589 TOKENIZER_FILE=" tekken.json"
8690 AUDIO_URL=" https://github.com/voxserv/audio_quality_testing_samples/raw/refs/heads/master/testaudio/16000/test01_20s.wav"
8791 AUDIO_FILE=" poem.wav"
@@ -93,7 +97,7 @@ case "$HF_MODEL" in
9397 RUNNER_PATH=" whisper"
9498 EXPECTED_OUTPUT=" Mr. Quilter"
9599 PREPROCESSOR=" whisper_preprocessor.pte"
96- TOKENIZER_URL=" https://huggingface.co/openai/whisper-small/resolve/main"
100+ TOKENIZER_URL=" https://huggingface.co/openai/whisper-small/resolve/main" # @lint-ignore
97101 TOKENIZER_FILE=" "
98102 AUDIO_URL=" "
99103 AUDIO_FILE=" output.wav"
@@ -105,7 +109,7 @@ case "$HF_MODEL" in
105109 RUNNER_PATH=" gemma3"
106110 EXPECTED_OUTPUT=" chip"
107111 PREPROCESSOR=" "
108- TOKENIZER_URL=" https://huggingface.co/google/gemma-3-4b-it/resolve/main"
112+ TOKENIZER_URL=" https://huggingface.co/google/gemma-3-4b-it/resolve/main" # @lint-ignore
109113 TOKENIZER_FILE=" "
110114 AUDIO_URL=" "
111115 AUDIO_FILE=" "
0 commit comments