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