from library.utils import setup_logging

bmaltais · Mar 2, 2024 · 80e82cf · 80e82cf
1 parent d612001
commit 80e82cf
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/tools/group_images.py b/tools/group_images.py
@@ -4,10 +4,12 @@
 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
@@ -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")