From b50710bf33762de6bd5f80a984bab97af4b98b2f Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sat, 14 Oct 2023 17:32:27 -0400 Subject: [PATCH] reduce debug log --- modules/shared.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/modules/shared.py b/modules/shared.py index 104412bc2..86293baee 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -97,6 +97,7 @@ class State: need_restart = False server_start = time.time() oom = False + debug_output = os.environ.get('SD_STATE_DEBUG', None) def skip(self): log.debug('Requested skip') @@ -147,14 +148,16 @@ class State: self.skipped = False self.textinfo = None self.time_start = time.time() - log.debug(f'State begin: {self.job}') + if self.debug_output: + log.debug(f'State begin: {self.job}') devices.torch_gc() def end(self): if self.time_start is None: # someone called end before being log.debug(f'Access state.end: {sys._getframe().f_back.f_code.co_name}') # pylint: disable=protected-access self.time_start = time.time() - log.debug(f'State end: {self.job} time={time.time() - self.time_start:.2f}s') + if self.debug_output: + log.debug(f'State end: {self.job} time={time.time() - self.time_start:.2f}s') self.job = "" self.job_count = 0 self.job_no = 0