From 3c6e8a556271c4b099e739e6428ec41ed4cad136 Mon Sep 17 00:00:00 2001 From: DepFA <35278260+dfaker@users.noreply.github.com> Date: Sun, 15 Jan 2023 17:50:56 +0000 Subject: [PATCH] Update shared.py --- modules/shared.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/modules/shared.py b/modules/shared.py index 3483db1c..ab8d6e84 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -171,6 +171,7 @@ class State: time_start = None need_restart = False server_start = None + is_hr_pass = False def skip(self): self.skipped = True @@ -196,6 +197,7 @@ class State: "job_no": self.job_no, "sampling_step": self.sampling_step, "sampling_steps": self.sampling_steps, + "is_hr_pass":self.is_hr_pass, } return obj @@ -214,12 +216,14 @@ class State: self.interrupted = False self.textinfo = None self.time_start = time.time() + self.is_hr_pass = False devices.torch_gc() def end(self): self.job = "" self.job_count = 0 + self.is_hr_pass = False devices.torch_gc()