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: