SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update model_split_percents for WhisperModelTest (#21922)

Browse Source 
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2023-03-03 14:35:08 +01:00
committed by GitHub
parent c82bd37169
commit fa9d2ad7ec
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/models/whisper/test_modeling_whisper.py
View File

@@ -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"