From 80e82cf8ee930332175bde0f776860394a5c6e12 Mon Sep 17 00:00:00 2001
From: Won-Kyu Park <wkpark@gmail.com>
Date: Sat, 2 Mar 2024 12:22:02 +0900
Subject: [PATCH] from library.utils import setup_logging

---
 tools/group_images.py | 6 ++++--
 tools/lcm_convert.py  | 7 +++++--
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/tools/group_images.py b/tools/group_images.py
index 19435f6..e92aef5 100644
--- a/tools/group_images.py
+++ b/tools/group_images.py
@@ -4,10 +4,12 @@ from PIL import Image, ImageOps
 import os
 import numpy as np
 
-from library.custom_logging import setup_logging
+from library.utils import setup_logging
+import logging
 
 # Set up logging
-log = setup_logging()
+setup_logging()
+log = logging.getLogger(__name__)
 
 class ImageProcessor:
 
diff --git a/tools/lcm_convert.py b/tools/lcm_convert.py
index 1fc39f6..96d0d77 100644
--- a/tools/lcm_convert.py
+++ b/tools/lcm_convert.py
@@ -1,12 +1,15 @@
 import argparse
 import torch
-from library.custom_logging import setup_logging
+import logging
+from library.utils import setup_logging
 from diffusers import StableDiffusionPipeline, StableDiffusionXLPipeline, LCMScheduler
 from library.sdxl_model_util import convert_diffusers_unet_state_dict_to_sdxl, sdxl_original_unet, save_stable_diffusion_checkpoint, _load_state_dict_on_device as load_state_dict_on_device
 from accelerate import init_empty_weights
 
 # Initialize logging
-logger = setup_logging()
+setup_logging()
+logger = logging.getLogger(__name__)
+
 
 def parse_command_line_arguments():
     argument_parser = argparse.ArgumentParser("lcm_convert")