From eca8d9cfb1a5b64c3abe2fb714ec0a76c89df4c7 Mon Sep 17 00:00:00 2001 From: Michael Goin Date: Wed, 19 Jul 2023 14:33:34 -0400 Subject: [PATCH 1/2] Update transformers/__init__.py warning --- src/deepsparse/transformers/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/deepsparse/transformers/__init__.py b/src/deepsparse/transformers/__init__.py index 7546184c27..fb60efa147 100644 --- a/src/deepsparse/transformers/__init__.py +++ b/src/deepsparse/transformers/__init__.py @@ -44,7 +44,7 @@ def _check_transformers_install(): if not getattr(_transformers, "NM_INTEGRATED", False): _LOGGER.warning( "The neuralmagic fork of transformers may not be installed. It can be " - f"installed via `pip install {nm_transformers}`" + f"installed via `pip install nm_transformers`" ) From 206bcf47a9dac681d2ce16f88d23bee677931d7c Mon Sep 17 00:00:00 2001 From: Michael Goin Date: Wed, 19 Jul 2023 14:36:03 -0400 Subject: [PATCH 2/2] Update __init__.py --- src/deepsparse/transformers/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/deepsparse/transformers/__init__.py b/src/deepsparse/transformers/__init__.py index fb60efa147..b434dd86cf 100644 --- a/src/deepsparse/transformers/__init__.py +++ b/src/deepsparse/transformers/__init__.py @@ -44,7 +44,7 @@ def _check_transformers_install(): if not getattr(_transformers, "NM_INTEGRATED", False): _LOGGER.warning( "The neuralmagic fork of transformers may not be installed. It can be " - f"installed via `pip install nm_transformers`" + "installed via `pip install nm_transformers`" )