diff --git a/convert-persimmon-to-gguf.py b/convert-persimmon-to-gguf.py index 7ab27ffb2..8fc91a7e5 100755 --- a/convert-persimmon-to-gguf.py +++ b/convert-persimmon-to-gguf.py @@ -73,10 +73,7 @@ def main(): parser.add_argument("--adept-inference-dir", type=str, help="path to adept-inference code directory") parser.add_argument("--verbose", action="store_true", help="increase output verbosity") args = parser.parse_args() - if args.verbose: - logging.basicConfig(level=logging.DEBUG) - else: - logging.basicConfig(level=logging.INFO) + logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO) sys.path.append(str(args.adept_inference_dir)) persimmon_model = torch.load(args.ckpt_path) hparams = persimmon_model['args'] diff --git a/ggml_vk_generate_shaders.py b/ggml_vk_generate_shaders.py index cd9b8dd96..1d9a0cc82 100644 --- a/ggml_vk_generate_shaders.py +++ b/ggml_vk_generate_shaders.py @@ -2775,10 +2775,7 @@ if __name__ == "__main__": args = parser.parse_args() - if args.verbose: - logging.basicConfig(level=logging.DEBUG) - else: - logging.basicConfig(level=logging.INFO) + logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO) if args.glslc: GLSLC = args.glslc diff --git a/gguf-py/scripts/gguf-convert-endian.py b/gguf-py/scripts/gguf-convert-endian.py index dc2b9dfbd..fdab4e72e 100755 --- a/gguf-py/scripts/gguf-convert-endian.py +++ b/gguf-py/scripts/gguf-convert-endian.py @@ -110,10 +110,7 @@ def main() -> None: args = parser.parse_args(None if len(sys.argv) > 1 else ["--help"]) - if args.verbose: - logging.basicConfig(level=logging.DEBUG) - else: - logging.basicConfig(level=logging.INFO) + logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO) logger.info(f'* Loading: {args.model}') reader = gguf.GGUFReader(args.model, 'r' if args.dry_run else 'r+') diff --git a/gguf-py/scripts/gguf-dump.py b/gguf-py/scripts/gguf-dump.py index f813a982c..7b0ef82c1 100755 --- a/gguf-py/scripts/gguf-dump.py +++ b/gguf-py/scripts/gguf-dump.py @@ -111,10 +111,7 @@ def main() -> None: args = parser.parse_args(None if len(sys.argv) > 1 else ["--help"]) - if args.verbose: - logging.basicConfig(level=logging.DEBUG) - else: - logging.basicConfig(level=logging.INFO) + logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO) if not args.json: logger.info(f'* Loading: {args.model}') diff --git a/gguf-py/scripts/gguf-set-metadata.py b/gguf-py/scripts/gguf-set-metadata.py index ed24dd75f..e35b651b8 100755 --- a/gguf-py/scripts/gguf-set-metadata.py +++ b/gguf-py/scripts/gguf-set-metadata.py @@ -84,10 +84,7 @@ def main() -> None: args = parser.parse_args(None if len(sys.argv) > 1 else ["--help"]) - if args.verbose: - logging.basicConfig(level=logging.DEBUG) - else: - logging.basicConfig(level=logging.INFO) + logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO) logger.info(f'* Loading: {args.model}') reader = GGUFReader(args.model, 'r' if args.dry_run else 'r+') diff --git a/tests/test-tokenizer-0-bpe.py b/tests/test-tokenizer-0-bpe.py index 731ae5e77..381332815 100644 --- a/tests/test-tokenizer-0-bpe.py +++ b/tests/test-tokenizer-0-bpe.py @@ -20,11 +20,7 @@ parser.add_argument("--fname-tok", help="path to a text file to tokenize") parser.add_argument("--verbose", action="store_true", help="increase output verbosity") args = parser.parse_args() -if args.verbose: - logging.basicConfig(level=logging.DEBUG) -else: - logging.basicConfig(level=logging.INFO) - +logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO) dir_tokenizer = args.dir_tokenizer tokenizer = AutoTokenizer.from_pretrained(dir_tokenizer) diff --git a/tests/test-tokenizer-0-spm.py b/tests/test-tokenizer-0-spm.py index 648b60908..223ab1bb6 100644 --- a/tests/test-tokenizer-0-spm.py +++ b/tests/test-tokenizer-0-spm.py @@ -21,10 +21,7 @@ parser.add_argument("--verbose", action="store_true", help="increase output verb args = parser.parse_args() -if args.verbose: - logging.basicConfig(level=logging.DEBUG) -else: - logging.basicConfig(level=logging.INFO) +logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO) dir_tokenizer = args.dir_tokenizer