fix: Removed duplicate field definitions in some classes (#31888)
Removed duplicate field definitions in classes.
This commit is contained in:
@@ -295,7 +295,6 @@ class FNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
|
||||
# Skip Tests
|
||||
test_pruning = False
|
||||
test_head_masking = False
|
||||
test_pruning = False
|
||||
|
||||
# TODO: Fix the failed tests
|
||||
def is_pipeline_test_to_skip(
|
||||
|
||||
@@ -258,7 +258,6 @@ class MambaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixi
|
||||
test_model_parallel = False
|
||||
test_pruning = False
|
||||
test_head_masking = False # Mamba does not have attention heads
|
||||
test_model_parallel = False
|
||||
pipeline_model_mapping = (
|
||||
{"feature-extraction": MambaModel, "text-generation": MambaForCausalLM} if is_torch_available() else {}
|
||||
)
|
||||
|
||||
@@ -298,7 +298,6 @@ class RecurrentGemmaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineT
|
||||
test_model_parallel = False
|
||||
test_pruning = False
|
||||
test_head_masking = False # RecurrentGemma does not have attention heads
|
||||
test_model_parallel = False
|
||||
|
||||
# Need to remove 0.9 in `test_cpu_offload`
|
||||
# This is because we are hitting edge cases with the causal_mask buffer
|
||||
|
||||
Reference in New Issue
Block a user