From f95563393cc55c37f5907196730bdc26f1b20ff5 Mon Sep 17 00:00:00 2001 From: brian khuu Date: Wed, 8 May 2024 15:28:56 +1000 Subject: [PATCH] compare-llama-bench.py: add missing basicConfig --- scripts/compare-llama-bench.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/scripts/compare-llama-bench.py b/scripts/compare-llama-bench.py index 3892fd25c..4436e4765 100755 --- a/scripts/compare-llama-bench.py +++ b/scripts/compare-llama-bench.py @@ -93,9 +93,12 @@ help_s = ( "specified values are averaged WITHOUT weighing by the --repetitions parameter of llama-bench." ) parser.add_argument("-s", "--show", help=help_s) +parser.add_argument("--verbose", action="store_true", help="increase output verbosity") known_args, unknown_args = parser.parse_known_args() +logging.basicConfig(level=logging.DEBUG if known_args.verbose else logging.INFO) + if unknown_args: logger.error(f"Received unknown args: {unknown_args}.") parser.print_help()