Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
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