From 57dd6652dfdead0e0efd1ebfad0cff29dea2221a Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Mon, 17 Jul 2023 19:30:53 +0000 Subject: [PATCH] fix compile logging --- modules/sd_hijack.py | 3 ++- modules/sd_models.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py index decefeb11..362285f4a 100644 --- a/modules/sd_hijack.py +++ b/modules/sd_hijack.py @@ -185,7 +185,8 @@ class StableDiffusionModelHijack: else: import torch._dynamo # pylint: disable=unused-import,redefined-outer-name log_level = logging.WARNING if opts.cuda_compile_verbose else logging.CRITICAL # pylint: disable=protected-access - torch._logging.set_logs(dynamo=log_level, aot=log_level, inductor=log_level) # pylint: disable=protected-access + if hasattr(torch, '_logging'): + torch._logging.set_logs(dynamo=log_level, aot=log_level, inductor=log_level) # pylint: disable=protected-access torch._dynamo.config.verbose = opts.cuda_compile_verbose # pylint: disable=protected-access torch._dynamo.config.suppress_errors = opts.cuda_compile_errors # pylint: disable=protected-access torch.backends.cudnn.benchmark = True diff --git a/modules/sd_models.py b/modules/sd_models.py index fcc43e62b..8f6c037e5 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -773,7 +773,8 @@ def load_diffuser(checkpoint_info=None, already_loaded_state_dict=None, timer=No else: import torch._dynamo # pylint: disable=unused-import,redefined-outer-name log_level = logging.WARNING if shared.opts.cuda_compile_verbose else logging.CRITICAL # pylint: disable=protected-access - torch._logging.set_logs(dynamo=log_level, aot=log_level, inductor=log_level) # pylint: disable=protected-access + if hasattr(torch, '_logging'): + torch._logging.set_logs(dynamo=log_level, aot=log_level, inductor=log_level) # pylint: disable=protected-access torch._dynamo.config.verbose = shared.opts.cuda_compile_verbose # pylint: disable=protected-access torch._dynamo.config.suppress_errors = shared.opts.cuda_compile_errors # pylint: disable=protected-access sd_model.unet = torch.compile(sd_model.unet, mode=shared.opts.cuda_compile_mode, fullgraph=shared.opts.cuda_compile_fullgraph) # pylint: disable=attribute-defined-outside-init