Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
609e6622
Unverified
Commit
609e6622
authored
Jun 19, 2024
by
amyeroberts
Committed by
GitHub
Jun 19, 2024
Browse files
Use self.config_tester.run_common_tests() (#31431)
* First testing updating config tests * Use run_common_tests
parent
7c71b61d
Changes
28
Show whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
116 additions
and
128 deletions
+116
-128
src/transformers/models/align/configuration_align.py
src/transformers/models/align/configuration_align.py
+1
-1
src/transformers/models/timm_backbone/configuration_timm_backbone.py
...rmers/models/timm_backbone/configuration_timm_backbone.py
+1
-1
tests/models/align/test_modeling_align.py
tests/models/align/test_modeling_align.py
+6
-7
tests/models/bit/test_modeling_bit.py
tests/models/bit/test_modeling_bit.py
+4
-7
tests/models/convnext/test_modeling_convnext.py
tests/models/convnext/test_modeling_convnext.py
+8
-7
tests/models/convnextv2/test_modeling_convnextv2.py
tests/models/convnextv2/test_modeling_convnextv2.py
+8
-7
tests/models/cpmant/test_modeling_cpmant.py
tests/models/cpmant/test_modeling_cpmant.py
+1
-6
tests/models/cvt/test_modeling_cvt.py
tests/models/cvt/test_modeling_cvt.py
+8
-7
tests/models/deformable_detr/test_modeling_deformable_detr.py
...s/models/deformable_detr/test_modeling_deformable_detr.py
+7
-7
tests/models/depth_anything/test_modeling_depth_anything.py
tests/models/depth_anything/test_modeling_depth_anything.py
+6
-8
tests/models/dinat/test_modeling_dinat.py
tests/models/dinat/test_modeling_dinat.py
+4
-7
tests/models/donut/test_modeling_donut_swin.py
tests/models/donut/test_modeling_donut_swin.py
+8
-7
tests/models/efficientnet/test_modeling_efficientnet.py
tests/models/efficientnet/test_modeling_efficientnet.py
+6
-7
tests/models/focalnet/test_modeling_focalnet.py
tests/models/focalnet/test_modeling_focalnet.py
+8
-7
tests/models/grounding_dino/test_modeling_grounding_dino.py
tests/models/grounding_dino/test_modeling_grounding_dino.py
+7
-7
tests/models/levit/test_modeling_levit.py
tests/models/levit/test_modeling_levit.py
+4
-7
tests/models/maskformer/test_modeling_maskformer_swin.py
tests/models/maskformer/test_modeling_maskformer_swin.py
+8
-7
tests/models/perceiver/test_modeling_perceiver.py
tests/models/perceiver/test_modeling_perceiver.py
+7
-7
tests/models/regnet/test_modeling_regnet.py
tests/models/regnet/test_modeling_regnet.py
+7
-7
tests/models/resnet/test_modeling_resnet.py
tests/models/resnet/test_modeling_resnet.py
+7
-7
No files found.
src/transformers/models/align/configuration_align.py
View file @
609e6622
...
@@ -193,7 +193,7 @@ class AlignVisionConfig(PretrainedConfig):
...
@@ -193,7 +193,7 @@ class AlignVisionConfig(PretrainedConfig):
hidden_act (`str` or `function`, *optional*, defaults to `"silu"`):
hidden_act (`str` or `function`, *optional*, defaults to `"silu"`):
The non-linear activation function (function or string) in each block. If string, `"gelu"`, `"relu"`,
The non-linear activation function (function or string) in each block. If string, `"gelu"`, `"relu"`,
`"selu", `"gelu_new"`, `"silu"` and `"mish"` are supported.
`"selu", `"gelu_new"`, `"silu"` and `"mish"` are supported.
hidde
m
_dim (`int`, *optional*, defaults to 1280):
hidde
n
_dim (`int`, *optional*, defaults to 1280):
The hidden dimension of the layer before the classification head.
The hidden dimension of the layer before the classification head.
pooling_type (`str` or `function`, *optional*, defaults to `"mean"`):
pooling_type (`str` or `function`, *optional*, defaults to `"mean"`):
Type of final pooling to be applied before the dense classification head. Available options are [`"mean"`,
Type of final pooling to be applied before the dense classification head. Available options are [`"mean"`,
...
...
src/transformers/models/timm_backbone/configuration_timm_backbone.py
View file @
609e6622
...
@@ -79,5 +79,5 @@ class TimmBackboneConfig(PretrainedConfig):
...
@@ -79,5 +79,5 @@ class TimmBackboneConfig(PretrainedConfig):
self
.
features_only
=
features_only
self
.
features_only
=
features_only
self
.
use_pretrained_backbone
=
use_pretrained_backbone
self
.
use_pretrained_backbone
=
use_pretrained_backbone
self
.
use_timm_backbone
=
True
self
.
use_timm_backbone
=
True
self
.
out_indices
=
out_indices
if
out_indices
is
not
None
else
(
-
1
,)
self
.
out_indices
=
out_indices
if
out_indices
is
not
None
else
[
-
1
]
self
.
freeze_batch_norm_2d
=
freeze_batch_norm_2d
self
.
freeze_batch_norm_2d
=
freeze_batch_norm_2d
tests/models/align/test_modeling_align.py
View file @
609e6622
...
@@ -147,16 +147,15 @@ class AlignVisionModelTest(ModelTesterMixin, unittest.TestCase):
...
@@ -147,16 +147,15 @@ class AlignVisionModelTest(ModelTesterMixin, unittest.TestCase):
def
setUp
(
self
):
def
setUp
(
self
):
self
.
model_tester
=
AlignVisionModelTester
(
self
)
self
.
model_tester
=
AlignVisionModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
AlignVisionConfig
,
has_text_modality
=
False
,
hidden_size
=
37
self
,
config_class
=
AlignVisionConfig
,
has_text_modality
=
False
,
hidden_size
=
37
,
common_properties
=
[
"num_channels"
,
"image_size"
],
)
)
def
test_config
(
self
):
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
@
unittest
.
skip
(
reason
=
"AlignVisionModel does not use inputs_embeds"
)
@
unittest
.
skip
(
reason
=
"AlignVisionModel does not use inputs_embeds"
)
def
test_inputs_embeds
(
self
):
def
test_inputs_embeds
(
self
):
...
...
tests/models/bit/test_modeling_bit.py
View file @
609e6622
...
@@ -173,15 +173,12 @@ class BitModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
...
@@ -173,15 +173,12 @@ class BitModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
def
setUp
(
self
):
def
setUp
(
self
):
self
.
model_tester
=
BitModelTester
(
self
)
self
.
model_tester
=
BitModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
BitConfig
,
has_text_modality
=
False
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
BitConfig
,
has_text_modality
=
False
,
common_properties
=
[
"num_channels"
]
)
def
test_config
(
self
):
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
@
unittest
.
skip
(
reason
=
"Bit does not output attentions"
)
@
unittest
.
skip
(
reason
=
"Bit does not output attentions"
)
def
test_attention_outputs
(
self
):
def
test_attention_outputs
(
self
):
...
...
tests/models/convnext/test_modeling_convnext.py
View file @
609e6622
...
@@ -183,15 +183,16 @@ class ConvNextModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase
...
@@ -183,15 +183,16 @@ class ConvNextModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase
def
setUp
(
self
):
def
setUp
(
self
):
self
.
model_tester
=
ConvNextModelTester
(
self
)
self
.
model_tester
=
ConvNextModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
ConvNextConfig
,
has_text_modality
=
False
,
hidden_size
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
ConvNextConfig
,
has_text_modality
=
False
,
hidden_size
=
37
,
common_properties
=
[
"num_channels"
,
"hidden_sizes"
],
)
def
test_config
(
self
):
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
@
unittest
.
skip
(
reason
=
"ConvNext does not use inputs_embeds"
)
@
unittest
.
skip
(
reason
=
"ConvNext does not use inputs_embeds"
)
def
test_inputs_embeds
(
self
):
def
test_inputs_embeds
(
self
):
...
...
tests/models/convnextv2/test_modeling_convnextv2.py
View file @
609e6622
...
@@ -191,15 +191,16 @@ class ConvNextV2ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCa
...
@@ -191,15 +191,16 @@ class ConvNextV2ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCa
def
setUp
(
self
):
def
setUp
(
self
):
self
.
model_tester
=
ConvNextV2ModelTester
(
self
)
self
.
model_tester
=
ConvNextV2ModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
ConvNextV2Config
,
has_text_modality
=
False
,
hidden_size
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
ConvNextV2Config
,
has_text_modality
=
False
,
hidden_size
=
37
,
common_properties
=
[
"hidden_sizes"
,
"num_channels"
],
)
def
test_config
(
self
):
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
@
unittest
.
skip
(
reason
=
"ConvNextV2 does not use inputs_embeds"
)
@
unittest
.
skip
(
reason
=
"ConvNextV2 does not use inputs_embeds"
)
def
test_inputs_embeds
(
self
):
def
test_inputs_embeds
(
self
):
...
...
tests/models/cpmant/test_modeling_cpmant.py
View file @
609e6622
...
@@ -151,12 +151,7 @@ class CpmAntModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
...
@@ -151,12 +151,7 @@ class CpmAntModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
CpmAntConfig
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
CpmAntConfig
)
def
test_config
(
self
):
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_common_properties
()
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
def
test_inputs_embeds
(
self
):
def
test_inputs_embeds
(
self
):
unittest
.
skip
(
"CPMAnt doesn't support input_embeds."
)(
self
.
test_inputs_embeds
)
unittest
.
skip
(
"CPMAnt doesn't support input_embeds."
)(
self
.
test_inputs_embeds
)
...
...
tests/models/cvt/test_modeling_cvt.py
View file @
609e6622
...
@@ -162,15 +162,16 @@ class CvtModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
...
@@ -162,15 +162,16 @@ class CvtModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
def
setUp
(
self
):
def
setUp
(
self
):
self
.
model_tester
=
CvtModelTester
(
self
)
self
.
model_tester
=
CvtModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
CvtConfig
,
has_text_modality
=
False
,
hidden_size
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
CvtConfig
,
has_text_modality
=
False
,
hidden_size
=
37
,
common_properties
=
[
"hidden_size"
,
"num_channels"
],
)
def
test_config
(
self
):
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
@
unittest
.
skip
(
reason
=
"Cvt does not output attentions"
)
@
unittest
.
skip
(
reason
=
"Cvt does not output attentions"
)
def
test_attention_outputs
(
self
):
def
test_attention_outputs
(
self
):
...
...
tests/models/deformable_detr/test_modeling_deformable_detr.py
View file @
609e6622
...
@@ -230,15 +230,15 @@ class DeformableDetrModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineT
...
@@ -230,15 +230,15 @@ class DeformableDetrModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineT
def
setUp
(
self
):
def
setUp
(
self
):
self
.
model_tester
=
DeformableDetrModelTester
(
self
)
self
.
model_tester
=
DeformableDetrModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
DeformableDetrConfig
,
has_text_modality
=
False
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
DeformableDetrConfig
,
has_text_modality
=
False
,
common_properties
=
[
"num_channels"
,
"d_model"
,
"encoder_attention_heads"
,
"decoder_attention_heads"
],
)
def
test_config
(
self
):
def
test_config
(
self
):
# we don't test common_properties and arguments_init as these don't apply for Deformable DETR
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
def
test_deformable_detr_model
(
self
):
def
test_deformable_detr_model
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
...
...
tests/models/depth_anything/test_modeling_depth_anything.py
View file @
609e6622
...
@@ -149,17 +149,15 @@ class DepthAnythingModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Tes
...
@@ -149,17 +149,15 @@ class DepthAnythingModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Tes
def
setUp
(
self
):
def
setUp
(
self
):
self
.
model_tester
=
DepthAnythingModelTester
(
self
)
self
.
model_tester
=
DepthAnythingModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
DepthAnythingConfig
,
has_text_modality
=
False
,
hidden_size
=
37
self
,
config_class
=
DepthAnythingConfig
,
has_text_modality
=
False
,
hidden_size
=
37
,
common_properties
=
[
"patch_size"
],
)
)
def
test_config
(
self
):
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained_subfolder
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
@
unittest
.
skip
(
reason
=
"Depth Anything with AutoBackbone does not have a base model and hence no input_embeddings"
)
@
unittest
.
skip
(
reason
=
"Depth Anything with AutoBackbone does not have a base model and hence no input_embeddings"
)
def
test_inputs_embeds
(
self
):
def
test_inputs_embeds
(
self
):
...
...
tests/models/dinat/test_modeling_dinat.py
View file @
609e6622
...
@@ -219,15 +219,12 @@ class DinatModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
...
@@ -219,15 +219,12 @@ class DinatModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
def
setUp
(
self
):
def
setUp
(
self
):
self
.
model_tester
=
DinatModelTester
(
self
)
self
.
model_tester
=
DinatModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
DinatConfig
,
embed_dim
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
DinatConfig
,
embed_dim
=
37
,
common_properties
=
[
"patch_size"
,
"num_channels"
]
)
def
test_config
(
self
):
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
def
test_model
(
self
):
def
test_model
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
...
...
tests/models/donut/test_modeling_donut_swin.py
View file @
609e6622
...
@@ -153,15 +153,16 @@ class DonutSwinModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCas
...
@@ -153,15 +153,16 @@ class DonutSwinModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCas
def
setUp
(
self
):
def
setUp
(
self
):
self
.
model_tester
=
DonutSwinModelTester
(
self
)
self
.
model_tester
=
DonutSwinModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
DonutSwinConfig
,
embed_dim
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
DonutSwinConfig
,
has_text_modality
=
False
,
embed_dim
=
37
,
common_properties
=
[
"image_size"
,
"patch_size"
,
"num_channels"
],
)
def
test_config
(
self
):
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
def
test_model
(
self
):
def
test_model
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
...
...
tests/models/efficientnet/test_modeling_efficientnet.py
View file @
609e6622
...
@@ -142,16 +142,15 @@ class EfficientNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Test
...
@@ -142,16 +142,15 @@ class EfficientNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Test
def
setUp
(
self
):
def
setUp
(
self
):
self
.
model_tester
=
EfficientNetModelTester
(
self
)
self
.
model_tester
=
EfficientNetModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
EfficientNetConfig
,
has_text_modality
=
False
,
hidden_size
=
37
self
,
config_class
=
EfficientNetConfig
,
has_text_modality
=
False
,
hidden_size
=
37
,
common_properties
=
[
"num_channels"
,
"image_size"
,
"hidden_dim"
],
)
)
def
test_config
(
self
):
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
@
unittest
.
skip
(
reason
=
"EfficientNet does not use inputs_embeds"
)
@
unittest
.
skip
(
reason
=
"EfficientNet does not use inputs_embeds"
)
def
test_inputs_embeds
(
self
):
def
test_inputs_embeds
(
self
):
...
...
tests/models/focalnet/test_modeling_focalnet.py
View file @
609e6622
...
@@ -250,15 +250,16 @@ class FocalNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase
...
@@ -250,15 +250,16 @@ class FocalNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase
def
setUp
(
self
):
def
setUp
(
self
):
self
.
model_tester
=
FocalNetModelTester
(
self
)
self
.
model_tester
=
FocalNetModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
FocalNetConfig
,
embed_dim
=
37
,
has_text_modality
=
False
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
FocalNetConfig
,
embed_dim
=
37
,
has_text_modality
=
False
,
common_properties
=
[
"image_size"
,
"patch_size"
,
"num_channels"
,
"hidden_sizes"
],
)
def
test_config
(
self
):
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
def
test_model
(
self
):
def
test_model
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
...
...
tests/models/grounding_dino/test_modeling_grounding_dino.py
View file @
609e6622
...
@@ -246,15 +246,15 @@ class GroundingDinoModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Tes
...
@@ -246,15 +246,15 @@ class GroundingDinoModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Tes
def
setUp
(
self
):
def
setUp
(
self
):
self
.
model_tester
=
GroundingDinoModelTester
(
self
)
self
.
model_tester
=
GroundingDinoModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
GroundingDinoConfig
,
has_text_modality
=
False
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
GroundingDinoConfig
,
has_text_modality
=
False
,
common_properties
=
[
"d_model"
,
"encoder_attention_heads"
,
"decoder_attention_heads"
],
)
def
test_config
(
self
):
def
test_config
(
self
):
# we don't test common_properties and arguments_init as these don't apply for Grounding DINO
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
def
test_model
(
self
):
def
test_model
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
...
...
tests/models/levit/test_modeling_levit.py
View file @
609e6622
...
@@ -190,15 +190,12 @@ class LevitModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
...
@@ -190,15 +190,12 @@ class LevitModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
def
setUp
(
self
):
def
setUp
(
self
):
self
.
model_tester
=
LevitModelTester
(
self
)
self
.
model_tester
=
LevitModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
LevitConfig
,
has_text_modality
=
False
,
hidden_size
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
LevitConfig
,
has_text_modality
=
False
,
common_properties
=
[
"image_size"
,
"num_channels"
]
)
def
test_config
(
self
):
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
@
unittest
.
skip
(
reason
=
"Levit does not use inputs_embeds"
)
@
unittest
.
skip
(
reason
=
"Levit does not use inputs_embeds"
)
def
test_inputs_embeds
(
self
):
def
test_inputs_embeds
(
self
):
...
...
tests/models/maskformer/test_modeling_maskformer_swin.py
View file @
609e6622
...
@@ -184,7 +184,13 @@ class MaskFormerSwinModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Te
...
@@ -184,7 +184,13 @@ class MaskFormerSwinModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Te
def
setUp
(
self
):
def
setUp
(
self
):
self
.
model_tester
=
MaskFormerSwinModelTester
(
self
)
self
.
model_tester
=
MaskFormerSwinModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
MaskFormerSwinConfig
,
embed_dim
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
MaskFormerSwinConfig
,
has_text_modality
=
False
,
embed_dim
=
37
,
common_properties
=
[
"image_size"
,
"patch_size"
,
"num_channels"
],
)
@
require_torch_multi_gpu
@
require_torch_multi_gpu
@
unittest
.
skip
(
@
unittest
.
skip
(
...
@@ -197,12 +203,7 @@ class MaskFormerSwinModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Te
...
@@ -197,12 +203,7 @@ class MaskFormerSwinModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.Te
pass
pass
def
test_config
(
self
):
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
def
test_model
(
self
):
def
test_model
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
()
...
...
tests/models/perceiver/test_modeling_perceiver.py
View file @
609e6622
...
@@ -315,7 +315,12 @@ class PerceiverModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCas
...
@@ -315,7 +315,12 @@ class PerceiverModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCas
def
setUp
(
self
):
def
setUp
(
self
):
self
.
model_tester
=
PerceiverModelTester
(
self
)
self
.
model_tester
=
PerceiverModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
PerceiverConfig
,
hidden_size
=
37
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
PerceiverConfig
,
hidden_size
=
37
,
common_properties
=
[
"d_model"
,
"num_self_attention_heads"
,
"num_cross_attention_heads"
],
)
def
_prepare_for_class
(
self
,
inputs_dict
,
model_class
,
return_labels
=
False
):
def
_prepare_for_class
(
self
,
inputs_dict
,
model_class
,
return_labels
=
False
):
inputs_dict
=
copy
.
deepcopy
(
inputs_dict
)
inputs_dict
=
copy
.
deepcopy
(
inputs_dict
)
...
@@ -344,12 +349,7 @@ class PerceiverModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCas
...
@@ -344,12 +349,7 @@ class PerceiverModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCas
return
inputs_dict
return
inputs_dict
def
test_config
(
self
):
def
test_config
(
self
):
# we don't test common_properties and arguments_init as these don't apply for Perceiver
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
def
test_for_masked_lm
(
self
):
def
test_for_masked_lm
(
self
):
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
(
model_class
=
PerceiverForMaskedLM
)
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs
(
model_class
=
PerceiverForMaskedLM
)
...
...
tests/models/regnet/test_modeling_regnet.py
View file @
609e6622
...
@@ -136,15 +136,15 @@ class RegNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
...
@@ -136,15 +136,15 @@ class RegNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
def
setUp
(
self
):
def
setUp
(
self
):
self
.
model_tester
=
RegNetModelTester
(
self
)
self
.
model_tester
=
RegNetModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
RegNetConfig
,
has_text_modality
=
False
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
RegNetConfig
,
has_text_modality
=
False
,
common_properties
=
[
"num_channels"
,
"hidden_sizes"
],
)
def
test_config
(
self
):
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
@
unittest
.
skip
(
reason
=
"RegNet does not use inputs_embeds"
)
@
unittest
.
skip
(
reason
=
"RegNet does not use inputs_embeds"
)
def
test_inputs_embeds
(
self
):
def
test_inputs_embeds
(
self
):
...
...
tests/models/resnet/test_modeling_resnet.py
View file @
609e6622
...
@@ -181,15 +181,15 @@ class ResNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
...
@@ -181,15 +181,15 @@ class ResNetModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
def
setUp
(
self
):
def
setUp
(
self
):
self
.
model_tester
=
ResNetModelTester
(
self
)
self
.
model_tester
=
ResNetModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
ResNetConfig
,
has_text_modality
=
False
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
ResNetConfig
,
has_text_modality
=
False
,
common_properties
=
[
"num_channels"
,
"hidden_sizes"
],
)
def
test_config
(
self
):
def
test_config
(
self
):
self
.
config_tester
.
create_and_test_config_to_json_string
()
self
.
config_tester
.
run_common_tests
()
self
.
config_tester
.
create_and_test_config_to_json_file
()
self
.
config_tester
.
create_and_test_config_from_and_save_pretrained
()
self
.
config_tester
.
create_and_test_config_with_num_labels
()
self
.
config_tester
.
check_config_can_be_init_without_params
()
self
.
config_tester
.
check_config_arguments_init
()
@
unittest
.
skip
(
reason
=
"ResNet does not use inputs_embeds"
)
@
unittest
.
skip
(
reason
=
"ResNet does not use inputs_embeds"
)
def
test_inputs_embeds
(
self
):
def
test_inputs_embeds
(
self
):
...
...
Prev
1
2
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment