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

Fix missing test in torch_job (#33593)

Browse Source 
fix missing tests

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2024-09-20 17:16:44 +02:00
committed by GitHub
parent 2fdb5e74cc
commit 31caf0b95f
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/generation/test_utils.py
View File

@@ -89,7 +89,6 @@ if is_torch_available():
from transformers.generation.utils import _speculative_sampling from transformers.generation.utils import _speculative_sampling
@pytest.mark.generate
class GenerationTesterMixin: class GenerationTesterMixin:
model_tester = None model_tester = None
all_generative_model_classes = () all_generative_model_classes = ()
@@ -2035,6 +2034,7 @@ class GenerationTesterMixin:
output_compiled = compiled_generate(model_inputs, generation_config=generation_config) output_compiled = compiled_generate(model_inputs, generation_config=generation_config)
self.assertListEqual(output_dynamic.tolist(), output_compiled.tolist()) self.assertListEqual(output_dynamic.tolist(), output_compiled.tolist())
@pytest.mark.generate
def test_generate_methods_with_num_logits_to_keep(self): def test_generate_methods_with_num_logits_to_keep(self):
for model_class in self.all_generative_model_classes: for model_class in self.all_generative_model_classes:
if "num_logits_to_keep" not in set(inspect.signature(model_class.forward).parameters.keys()): if "num_logits_to_keep" not in set(inspect.signature(model_class.forward).parameters.keys()):
@@ -2063,6 +2063,7 @@ class GenerationTesterMixin:
) )
self.assertEqual(with_all_logits.tolist(), without_all_logits.tolist()) self.assertEqual(with_all_logits.tolist(), without_all_logits.tolist())
@pytest.mark.generate
@is_flaky() # assisted generation tests are flaky (minor fp ops differences) @is_flaky() # assisted generation tests are flaky (minor fp ops differences)
def test_assisted_decoding_with_num_logits_to_keep(self): def test_assisted_decoding_with_num_logits_to_keep(self):
for model_class in self.all_generative_model_classes: for model_class in self.all_generative_model_classes: