Fix helper

Xilinx · Jan 29, 2025 · d9e4dcf · d9e4dcf
1 parent 60f1f0b
commit d9e4dcf
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/brevitas_examples/llm/main.py b/src/brevitas_examples/llm/main.py
@@ -783,7 +783,9 @@ def parse_args(args, override_defaults={}):
         type=str,
         default=None,
         choices=['per_tensor', 'per_row', 'per_group'],
-        help='Granularity for scales/zero-point of inputs. Default: per_tensor.')
+        help=
+        'Granularity for scales/zero-point of KV cache. If not set, it will use input-quant-granularity. Default: %(default)s'
+    )
     parser.add_argument(
         '--input-group-size',
         type=int,