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

7
tests/generation/test_utils.py
View File

@@ -1359,13 +1359,8 @@ class GenerationTesterMixin:
) )
# Sample constraints # Sample constraints
if not input_ids.dtype == torch.float32:
min_id = torch.min(input_ids) + 3
max_id = torch.max(input_ids)
else:
# otherwise this throws an error for Speech2TextModel since its inputs are floating points
min_id = 3 min_id = 3
max_id = 100 max_id = model.config.vocab_size
force_tokens = torch.randint(min_id, max_id, (1, 2)).tolist()[0] force_tokens = torch.randint(min_id, max_id, (1, 2)).tolist()[0]
constraints = [ constraints = [
PhrasalConstraint(force_tokens), PhrasalConstraint(force_tokens),

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): def test_generate_without_input_ids(self):
pass pass
@unittest.skip("Skip while we investigate while it's failing.")
def test_constrained_beam_search_generate_dict_output(self):
pass
@staticmethod @staticmethod
def _get_encoder_outputs( def _get_encoder_outputs(
model, input_ids, attention_mask, output_attentions=None, output_hidden_states=None, num_interleave=1 model, input_ids, attention_mask, output_attentions=None, output_hidden_states=None, num_interleave=1