Skip to content

GitLab

Unverified Commit fa9d2ad7 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Update `model_split_percents` for `WhisperModelTest` (#21922) 


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent c82bd371
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tests/models/whisper/test_modeling_whisper.py
View file @ fa9d2ad7
...@@ -285,7 +285,8 @@ class WhisperModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMi ...@@ -285,7 +285,8 @@ class WhisperModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMi
test_pruning = False test_pruning = False
test_missing_keys = False test_missing_keys = False
# Needs higher percentages after model tester's vocab_size is changed to 200 (PR #21222) # 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" input_name = "input_features"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment