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

Generate: Fix flaky indexing error in test_constrained_beam_search_generate_dict_output (#21561)

Browse Source
This commit is contained in:
Joao Gante
2023-02-13 15:12:07 +00:00
committed by GitHub
parent 93ed89bf40
commit 24273268b7
2 changed files with 2 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -632,10 +632,6 @@ class WhisperModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCas
def test_generate_without_input_ids(self):
pass
@unittest.skip("Skip while we investigate while it's failing.")
def test_constrained_beam_search_generate_dict_output(self):
pass
@staticmethod
def _get_encoder_outputs(
model, input_ids, attention_mask, output_attentions=None, output_hidden_states=None, num_interleave=1