| # Copyright (c) Facebook, Inc. and its affiliates. | |
| # | |
| # This source code is licensed under the MIT license found in the | |
| # LICENSE file in the root directory of this source tree. | |
| import importlib | |
| import os | |
| # automatically import any Python files in the models/ directory | |
| models_dir = os.path.dirname(__file__) | |
| for file in os.listdir(models_dir): | |
| path = os.path.join(models_dir, file) | |
| if ( | |
| not file.startswith("_") | |
| and not file.startswith(".") | |
| and (file.endswith(".py") or os.path.isdir(path)) | |
| ): | |
| model_name = file[: file.find(".py")] if file.endswith(".py") else file | |
| module = importlib.import_module("fairseq.model_parallel.models." + model_name) | |