fixing the library name - auto-gptq

it was a bit confusing to use `auto_gptq` while the library installation from pypi refers to `auto-gptq` https://github.com/PanQiWei/AutoGPTQ The change should be helpful if the `ImportError` is raised
dottxt-ai · Nov 27, 2023 · dccdb23 · dccdb23
1 parent 027b29b
commit dccdb23
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/outlines/models/gptq.py b/outlines/models/gptq.py
@@ -13,7 +13,7 @@ def gptq(
         from auto_gptq import AutoGPTQForCausalLM
     except ImportError:
         raise ImportError(
-            "The `auto_gptq` library needs to be installed in order to use `AutoGPTQ` models."
+            "The `auto-gptq` library needs to be installed in order to use `AutoGPTQ` models."
         )
 
     if device is not None: