add get_logger function

llava/conversation.py

@@ -6,7 +6,6 @@ import base64
 from io import BytesIO
 from PIL import Image
 from transformers import AutoTokenizer
-import utils
 
 class SeparatorStyle(Enum):
     """Different separator style."""

llava/model/language_model/llava_llama.py

@@ -30,8 +30,9 @@ import sys
 import os
 sys.path.append(os.path.abspath("."))
 sys.path.append(os.path.abspath("../.."))
-import utils
-logger= utils.get_logger(__name__)
+
+from llava.utils import get_logger
+logger= get_logger(__name__)
 
 class LlavaConfig(LlamaConfig):
     model_type = "llava_llama"

llava/train/train.py

@@ -36,8 +36,8 @@ from llava.model import *
 from llava.mm_utils import process_anyres_image, tokenizer_image_token
 
 from PIL import Image
-import utils
-logger = utils.get_logger(__name__)
+from llava.utils import get_logger
+logger= get_logger(__name__)
 
 local_rank = None
 

llava/utils.py

@@ -13,6 +13,18 @@ moderation_msg = "YOUR INPUT VIOLATES OUR CONTENT MODERATION GUIDELINES. PLEASE
 
 handler = None
 
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.INFO)
+logging.basicConfig(
+    format="[%(levelname)s|%(filename)s:%(lineno)s] %(asctime)s >> %(message)s"
+)
+
+
+def get_logger(name):
+    logger = logging.getLogger(name)
+    logger.setLevel(logging.INFO)
+    return logger
+
 
 def build_logger(logger_name, logger_filename):
     global handler