Tests: move generate tests to the right mixin and delete redundant tests (#34464)

* tmp commit

* tmp commit

* cull overwrites of deleted tests

* typo

* more specific docstring

* make fixup

* parameterize at the top?

* correction

* more deletions :D

* tmp commit

* for VLMs too

* fix _check_outputs

* test nit

* make fixup

* fix another flaky

* test_generate_from_inputs_embeds -- handle missing attention mask
This commit is contained in:
Joao Gante
2024-10-30 10:59:08 +00:00
committed by GitHub
parent 913330ca9f
commit 8a734ea2c3
46 changed files with 265 additions and 2348 deletions

View File

@@ -21,7 +21,6 @@ import unittest
import numpy as np
from datasets import Audio, load_dataset
from packaging import version
from parameterized import parameterized
from pytest import mark
@@ -745,22 +744,6 @@ class MimiModelTest(ModelTesterMixin, unittest.TestCase):
def test_sdpa_can_compile_dynamic(self):
pass
# For now, Let's focus only on GPU for `torch.compile`
@slow
@require_torch_gpu
def test_torch_compile(self):
if version.parse(torch.__version__) < version.parse("2.3"):
self.skipTest(reason="This test requires torch >= 2.3 to run.")
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
n_iter = 3
for model_class in self.all_model_classes:
model = model_class(config).to(torch_device)
model.forward = torch.compile(model.forward)
for i in range(n_iter):
_ = model(inputs_dict["input_values"].to(torch_device))
@is_flaky()
def test_batching_equivalence(self):
super().test_batching_equivalence()