notquitenothing
/
auto_sd_docker
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update shared.py

This commit is contained in:
DepFA 2023-01-15 17:50:56 +00:00 committed by GitHub
parent 205991df78
commit 3c6e8a5562
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/shared.py
View File

@ -171,6 +171,7 @@ class State:
time_start = None time_start = None
need_restart = False need_restart = False
server_start = None server_start = None
is_hr_pass = False
def skip(self): def skip(self):
self.skipped = True self.skipped = True
@ -196,6 +197,7 @@ class State:
"job_no": self.job_no, "job_no": self.job_no,
"sampling_step": self.sampling_step, "sampling_step": self.sampling_step,
"sampling_steps": self.sampling_steps, "sampling_steps": self.sampling_steps,
"is_hr_pass":self.is_hr_pass,
} }
return obj return obj
@ -214,12 +216,14 @@ class State:
self.interrupted = False self.interrupted = False
self.textinfo = None self.textinfo = None
self.time_start = time.time() self.time_start = time.time()
self.is_hr_pass = False
devices.torch_gc() devices.torch_gc()
def end(self): def end(self):
self.job = "" self.job = ""
self.job_count = 0 self.job_count = 0
self.is_hr_pass = False
devices.torch_gc() devices.torch_gc()