microsoft--unilm
wehub-resource-sync
33 行
1.3 KiB
Python
33 行
1.3 KiB
Python
# import argparse
|
|
# import importlib
|
|
# import os
|
|
# # from fairseq.models import import_models
|
|
# from fairseq.models import MODEL_REGISTRY, ARCH_MODEL_INV_REGISTRY
|
|
#
|
|
# # models_dir = os.path.dirname(__file__)
|
|
# # import_models(models_dir, "unilm.models")
|
|
#
|
|
#
|
|
# 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("unilm.models." + model_name)
|
|
#
|
|
# # extra `model_parser` for sphinx
|
|
# if model_name in MODEL_REGISTRY:
|
|
# parser = argparse.ArgumentParser(add_help=False)
|
|
# group_archs = parser.add_argument_group("Named architectures")
|
|
# group_archs.add_argument(
|
|
# "--arch", choices=ARCH_MODEL_INV_REGISTRY[model_name]
|
|
# )
|
|
# group_args = parser.add_argument_group("Additional command-line arguments")
|
|
# MODEL_REGISTRY[model_name].add_args(group_args)
|
|
# globals()[model_name + "_parser"] = parser
|
|
|
|
from .unigpt import UniGPTmodel |