CI: avoid human error, automatically infer generative models (#33212)
* tmp commit * move tests to the right class * remove ALL all_generative_model_classes = ... * skip tf roberta * skip InstructBlipForConditionalGenerationDecoderOnlyTest * videollava * reduce diff * reduce diff * remove on vlms * fix a few more * manual rebase bits * more manual rebase * remove all manual generative model class test entries * fix up to ernie * a few more removals * handle remaining cases * recurrent gemma * it's better here * make fixup * tf idefics is broken * tf bert + generate is broken * don't touch tf :() * don't touch tf :( * make fixup * better comments for test skips * revert tf changes * remove empty line removal * one more * missing one
This commit is contained in:
@@ -328,7 +328,6 @@ class BloomModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
|
||||
else ()
|
||||
)
|
||||
|
||||
all_generative_model_classes = (BloomForCausalLM,) if is_torch_available() else ()
|
||||
pipeline_model_mapping = (
|
||||
{
|
||||
"feature-extraction": BloomModel,
|
||||
|
||||
@@ -171,7 +171,6 @@ class FlaxBloomModelTester:
|
||||
@require_flax
|
||||
class FlaxBloomModelTest(FlaxModelTesterMixin, unittest.TestCase, FlaxGenerationTesterMixin):
|
||||
all_model_classes = (FlaxBloomModel, FlaxBloomForCausalLM) if is_flax_available() else ()
|
||||
all_generative_model_classes = () if is_flax_available() else ()
|
||||
|
||||
def setUp(self):
|
||||
self.model_tester = FlaxBloomModelTester(self)
|
||||
@@ -199,7 +198,6 @@ class FlaxBloomModelTest(FlaxModelTesterMixin, unittest.TestCase, FlaxGeneration
|
||||
@require_flax
|
||||
class FlaxBloomGenerationTest(unittest.TestCase):
|
||||
all_model_classes = (FlaxBloomForCausalLM,) if is_flax_available() else ()
|
||||
all_generative_model_classes = () if is_flax_available() else ()
|
||||
|
||||
def setUp(self):
|
||||
self.model_id = "bigscience/bloom-560m"
|
||||
|
||||
Reference in New Issue
Block a user