From 0d263597953810ed2e654f13587bab6931486b95 Mon Sep 17 00:00:00 2001
From: DKnight54 <126916963+DKnight54@users.noreply.github.com>
Date: Thu, 30 Jan 2025 02:41:31 +0800
Subject: [PATCH] Update gen_img.py

---
 gen_img.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/gen_img.py b/gen_img.py
index 68bee2ec..1d74b8b5 100644
--- a/gen_img.py
+++ b/gen_img.py
@@ -1485,7 +1485,11 @@ class ListPrompter:
 
 
 def main(args):
-    logging.basicConfig()
+    logging.basicConfig(
+        format="%(asctime)s - %(levelname)s - %(name)s - %(message)s",
+        datefmt="%m/%d/%Y %H:%M:%S",
+        level=logging.INFO,
+    )
     setup_logging(args)
     if args.fp16:
         dtype = torch.float16