set default to 'values'

Signed-off-by: shanjiaz <[email protected]>

Author: shanjiaz

src/llmcompressor/modifiers/quantization/calibration.py

@@ -250,10 +250,11 @@ def calibrate_kv_cache_input_hook(
     kv_cache = getattr(module, "kv_cache")
     if not hasattr(module, "_past_kv_name"):
         # Determine which past KV parameter name to use once and cache it
+        # TODO: Find a better place to cache this
         module._past_kv_name = (
-            "past_key_values"  # transformers#39956
-            if "past_key_values" in inspect.signature(module.forward).parameters
-            else "past_key_value"
+            "past_key_value"  # transformers#39956
+            if "past_key_value" in inspect.signature(module.forward).parameters
+            else "past_key_values"
         )
 
     kwargs[module._past_kv_name] = kv_cache