Spaces:
Paused
Paused
| # coding=utf-8 | |
| # Copyright 2023 The HuggingFace Inc. team. | |
| # | |
| # Licensed under the Apache License, Version 2.0 (the "License"); | |
| # you may not use this file except in compliance with the License. | |
| # You may obtain a copy of the License at | |
| # | |
| # http://www.apache.org/licenses/LICENSE-2.0 | |
| # | |
| # Unless required by applicable law or agreed to in writing, software | |
| # distributed under the License is distributed on an "AS IS" BASIS, | |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
| # See the License for the specific language governing permissions and | |
| # limitations under the License. | |
| import glob | |
| import os | |
| from get_test_info import get_tester_classes | |
| if __name__ == "__main__": | |
| failures = [] | |
| pattern = os.path.join("tests", "models", "**", "test_modeling_*.py") | |
| test_files = glob.glob(pattern) | |
| # TODO: deal with TF/Flax too | |
| test_files = [ | |
| x for x in test_files if not (x.startswith("test_modeling_tf_") or x.startswith("test_modeling_flax_")) | |
| ] | |
| for test_file in test_files: | |
| tester_classes = get_tester_classes(test_file) | |
| for tester_class in tester_classes: | |
| # A few tester classes don't have `parent` parameter in `__init__`. | |
| # TODO: deal this better | |
| try: | |
| tester = tester_class(parent=None) | |
| except Exception: | |
| continue | |
| if hasattr(tester, "get_config"): | |
| config = tester.get_config() | |
| for k, v in config.to_dict().items(): | |
| if isinstance(v, int): | |
| target = None | |
| if k in ["vocab_size"]: | |
| target = 100 | |
| elif k in ["max_position_embeddings"]: | |
| target = 128 | |
| elif k in ["hidden_size", "d_model"]: | |
| target = 40 | |
| elif k == ["num_layers", "num_hidden_layers", "num_encoder_layers", "num_decoder_layers"]: | |
| target = 5 | |
| if target is not None and v > target: | |
| failures.append( | |
| f"{tester_class.__name__} will produce a `config` of type `{config.__class__.__name__}`" | |
| f' with config["{k}"] = {v} which is too large for testing! Set its value to be smaller' | |
| f" than {target}." | |
| ) | |
| if len(failures) > 0: | |
| raise Exception(f"There were {len(failures)} failures:\n" + "\n".join(failures)) | |