From 0bb8a2c2878aec41f92b0af4a2ef892d11dbfa61 Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Thu, 30 Jan 2025 02:26:55 +0800
Subject: [PATCH] Test logging fix

---
 gen_img.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/gen_img.py b/gen_img.py
index ee106d14..68bee2ec 100644
--- a/gen_img.py
+++ b/gen_img.py
@@ -63,7 +63,7 @@ from networks.control_net_lllite import ControlNetLLLite
 from library.utils import GradualLatent, EulerAncestralDiscreteSchedulerGL
 from library.utils import setup_logging, add_logging_arguments
 
-setup_logging()
+
 import logging
 
 logger = logging.getLogger(__name__)
@@ -1485,6 +1485,8 @@ class ListPrompter:
 
 
 def main(args):
+    logging.basicConfig()
+    setup_logging(args)
     if args.fp16:
         dtype = torch.float16
     elif args.bf16: