facebookresearch/fairseq
1
1
Fork 0
mirror of https://github.com/facebookresearch/fairseq.git synced 2024-09-11 17:25:31 +03:00
Code Issues Packages Projects Releases Wiki Activity

Simplify hubconf

Browse Source 
Summary: Pull Request resolved: https://github.com/fairinternal/fairseq-py/pull/758

Differential Revision: D16418932

Pulled By: myleott

fbshipit-source-id: 59f005164b61b9fa712922eeb23525f7eec38f38
This commit is contained in:
Myle Ott 2019-07-22 08:11:58 -07:00 committed by Facebook Github Bot
parent 906411da12
commit 51ba35217d
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
hubconf.py
View File

@ -5,6 +5,8 @@
# the root directory of this source tree. An additional grant of patent rights
# can be found in the PATENTS file in the same directory.
import functools
from fairseq.models import MODEL_REGISTRY
@ -18,5 +20,11 @@ dependencies = [
]
for model, cls in MODEL_REGISTRY.items():
globals()[model] = cls.from_pretrained
for model_type, _cls in MODEL_REGISTRY.items():
for model_name in _cls.hub_models().keys():
globals()[model_name] = functools.partial(
_cls.from_pretrained,
model_name_or_path=model_name,
)
# to simplify the interface we only expose named models
#globals()[model_type] = _cls.from_pretrained