From 4a0a6bf6ea1a05247ef75bbadd299aa96ac3b491 Mon Sep 17 00:00:00 2001 From: Benjamin Date: Wed, 26 Jan 2022 17:22:22 -0500 Subject: [PATCH] add flag to signal NM integration is active --- src/transformers/__init__.py | 3 +++ 1 file changed, 3 insertions(+) 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__)