fix: Removed duplicate field definitions in some classes (#31888)

Removed duplicate field definitions in classes.
This commit is contained in:
Sai-Suraj-27
2024-07-10 18:16:31 +05:30
committed by GitHub
parent 9d98706b3f
commit da79b18087
8 changed files with 0 additions and 14 deletions

View File

@@ -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(

View File

@@ -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 {}
)

View File

@@ -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