two converter py files needs basicConfig() added

ggerganov · May 4, 2024 · 35f98ff · 35f98ff
1 parent a2ac89d
commit 35f98ff
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/convert-hf-to-gguf-update.py b/convert-hf-to-gguf-update.py
@@ -1,3 +1,5 @@
+#!/usr/bin/env python3
+
 # This script downloads the tokenizer models of the specified models from Huggingface and
 # generates the get_vocab_base_pre() function for convert-hf-to-gguf.py
 #
@@ -32,7 +34,7 @@
 from transformers import AutoTokenizer
 
 logger = logging.getLogger("convert-hf-to-gguf-update")
-
+logging.basicConfig(level=logging.INFO)
 
 class TOKENIZER_TYPE(IntEnum):
     SPM = auto()

diff --git a/convert-lora-to-ggml.py b/convert-lora-to-ggml.py
@@ -17,6 +17,7 @@
 import gguf
 
 logger = logging.getLogger("lora-to-gguf")
+logging.basicConfig(level=logging.INFO)
 
 NUMPY_TYPE_TO_FTYPE: dict[str, int] = {"float32": 0, "float16": 1}