fix error

huggingface · Jul 2, 2024 · e1a2a3e · e1a2a3e
1 parent b2ac4e9
commit e1a2a3e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/transformers/generation/utils.py b/src/transformers/generation/utils.py
@@ -1762,8 +1762,8 @@ def generate(
 
                 if cache_config.backend == "quanto" and not is_optimum_quanto_available():
                     raise ImportError(
-                        "You need to install `quanto` in order to use KV cache quantization with quanto backend. "
-                        "Please install it via  with `pip install quanto`"
+                        "You need to install `optimum-quanto` in order to use KV cache quantization with quanto backend. "
+                        "Please install it via  with `pip install optimum-quanto`"
                     )
                 elif cache_config.backend == "HQQ" and not is_hqq_available():
                     raise ImportError(