From c91dcdf2a41c6ca024f0768ea3465b266cefe3b1 Mon Sep 17 00:00:00 2001 From: teleprint-me <77757836+teleprint-me@users.noreply.github.com> Date: Thu, 23 May 2024 22:58:03 -0400 Subject: [PATCH] refactor: Add fixes for logging --- gguf-py/scripts/hub-vocab.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/gguf-py/scripts/hub-vocab.py b/gguf-py/scripts/hub-vocab.py index b7366739c..c300081e1 100644 --- a/gguf-py/scripts/hub-vocab.py +++ b/gguf-py/scripts/hub-vocab.py @@ -18,7 +18,7 @@ if ( from gguf.constants import MODEL_ARCH, MODEL_ARCH_NAMES from gguf.huggingface_hub import HFVocabRequest -logger = logging.getLogger("gguf-gen-pre") +logger = logging.getLogger(Path(__file__).stem) def main(): @@ -45,6 +45,11 @@ def main(): ) args = parser.parse_args() + if args.verbose: + logging.basicConfig(level=logging.DEBUG) + else: + logging.basicConfig(level=logging.INFO) + vocab_request = HFVocabRequest(args.auth_token, args.model_path, logger) vocab_type = vocab_request.get_vocab_enum(args.vocab_type) vocab_request.get_all_vocab_files(args.model_repo, vocab_type)