VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-17 17:24:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Test logging fix

Browse Source
This commit is contained in:
DKnight54
2025-01-30 02:26:55 +08:00
committed by GitHub
parent 9ae34ac527
commit 0bb8a2c287
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
gen_img.py
View File

@@ -63,7 +63,7 @@ from networks.control_net_lllite import ControlNetLLLite
from library.utils import GradualLatent, EulerAncestralDiscreteSchedulerGL from library.utils import GradualLatent, EulerAncestralDiscreteSchedulerGL
from library.utils import setup_logging, add_logging_arguments from library.utils import setup_logging, add_logging_arguments
setup_logging()
import logging import logging
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@@ -1485,6 +1485,8 @@ class ListPrompter:
def main(args): def main(args):
logging.basicConfig()
setup_logging(args)
if args.fp16: if args.fp16:
dtype = torch.float16 dtype = torch.float16
elif args.bf16: elif args.bf16: