mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-15 16:39:42 +00:00
Update gen_img.py
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user