Update gen_img.py

This commit is contained in:
DKnight54
2025-01-30 02:41:31 +08:00
committed by GitHub
parent 99bc4b6f48
commit 0d26359795

View File

@@ -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