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:
@@ -212,7 +212,6 @@ def sdpa_kernel(enable_flash, enable_math, enable_mem_efficient):
|
||||
class ModelTesterMixin:
|
||||
model_tester = None
|
||||
all_model_classes = ()
|
||||
all_generative_model_classes = ()
|
||||
fx_compatible = False
|
||||
test_torchscript = True
|
||||
test_pruning = True
|
||||
@@ -230,6 +229,10 @@ class ModelTesterMixin:
|
||||
_is_composite = False
|
||||
model_split_percents = [0.5, 0.7, 0.9]
|
||||
|
||||
@property
|
||||
def all_generative_model_classes(self):
|
||||
return tuple(model_class for model_class in self.all_model_classes if model_class.can_generate())
|
||||
|
||||
def _prepare_for_class(self, inputs_dict, model_class, return_labels=False):
|
||||
inputs_dict = copy.deepcopy(inputs_dict)
|
||||
if model_class.__name__ in get_values(MODEL_FOR_MULTIPLE_CHOICE_MAPPING_NAMES):
|
||||
|
||||
Reference in New Issue
Block a user