diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py index 26be362c5dab..b80633386607 100755 --- a/src/transformers/__init__.py +++ b/src/transformers/__init__.py @@ -2779,6 +2779,9 @@ class _LazyModule(_BaseLazyModule): __file__ = globals()["__file__"] __path__ = [os.path.dirname(__file__)] + # flag to signal NM integration is active + NM_INTEGRATED = True + def _get_module(self, module_name: str): return importlib.import_module("." + module_name, self.__name__)