Allow check_transformers_install to warn and pass if NM fork isn't us…

…ed (#1101) * Allow check_transformers_install to warn and pass if NM fork isn't used * . * Update src/deepsparse/transformers/__init__.py Co-authored-by: Benjamin Fineran <bfineran@users.noreply.github.com> --------- Co-authored-by: Benjamin Fineran <bfineran@users.noreply.github.com>
neuralmagic · Jul 6, 2023 · d8eafaf · d8eafaf
1 parent c6b17f9
commit d8eafaf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/deepsparse/transformers/__init__.py b/src/deepsparse/transformers/__init__.py
@@ -41,9 +41,9 @@
 def _check_transformers_install():
     import transformers as _transformers
 
-    if not _transformers.NM_INTEGRATED:
+    if not getattr(_transformers, "NM_INTEGRATED", False):
         _LOGGER.warning(
-            "the neuralmagic fork of transformers may not be installed. it can be "
+            "The neuralmagic fork of transformers may not be installed. It can be "
             f"installed via `pip install {nm_transformers}`"
         )