From fa9d2ad7ec0b41e6ac646cde507b747e7688a7c0 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Fri, 3 Mar 2023 14:35:08 +0100 Subject: [PATCH] Update `model_split_percents` for `WhisperModelTest` (#21922) Co-authored-by: ydshieh --- tests/models/whisper/test_modeling_whisper.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/models/whisper/test_modeling_whisper.py b/tests/models/whisper/test_modeling_whisper.py index 1b7b731a41..77bd01a010 100644 --- a/tests/models/whisper/test_modeling_whisper.py +++ b/tests/models/whisper/test_modeling_whisper.py @@ -285,7 +285,8 @@ class WhisperModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMi test_pruning = False test_missing_keys = False # Needs higher percentages after model tester's vocab_size is changed to 200 (PR #21222) - model_split_percents = [0.8, 0.9] + # `0.5` is for `test_disk_offload` (which also works for `test_model_parallelism`) + model_split_percents = [0.5, 0.8, 0.9] input_name = "input_features"