add flag to signal NM integration is active (#32)

neuralmagic · Jan 26, 2022 · 410f9db · 410f9db
1 parent c7b33f0
commit 410f9db
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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__)