diff options
| author | Aarni Koskela <akx@iki.fi> | 2023-06-30 10:11:31 +0000 |
|---|---|---|
| committer | Aarni Koskela <akx@iki.fi> | 2023-07-03 08:03:41 +0000 |
| commit | f44feb6a10aacc6a5ff4c9275fba2546b2858935 (patch) | |
| tree | a6c4896cd07020a7db5d0fa7cfb8eefbe5351bb4 /modules/shared.py | |
| parent | b70001e618d0f0015273e1313cc7ebe3002a4510 (diff) | |
| download | stable-diffusion-webui-gfx803-f44feb6a10aacc6a5ff4c9275fba2546b2858935.tar.gz stable-diffusion-webui-gfx803-f44feb6a10aacc6a5ff4c9275fba2546b2858935.tar.bz2 stable-diffusion-webui-gfx803-f44feb6a10aacc6a5ff4c9275fba2546b2858935.zip | |
Add job argument to State.begin()
Diffstat (limited to 'modules/shared.py')
| -rw-r--r-- | modules/shared.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/shared.py b/modules/shared.py index 203ee1b9..7df2879c 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -173,7 +173,7 @@ class State: return obj
- def begin(self):
+ def begin(self, job: str = "(unknown)"):
self.sampling_step = 0
self.job_count = -1
self.processing_has_refined_job_count = False
@@ -187,7 +187,7 @@ class State: self.interrupted = False
self.textinfo = None
self.time_start = time.time()
-
+ self.job = job
devices.torch_gc()
def end(self):
|