diff options
| author | Fampai <> | 2022-10-31 13:54:51 +0000 |
|---|---|---|
| committer | Fampai <> | 2022-10-31 13:54:51 +0000 |
| commit | 3b0127e698a2eeb913437bce0b25b478fb06ff11 (patch) | |
| tree | e0894e49eb0d7609b12d4e4f3a71fd979830b29c /modules/shared.py | |
| parent | 006756f9cd6258eae418e9209cfc13f940ec53e1 (diff) | |
| parent | 9b384dfb5c05129f50cc3f0262f89e8b788e5cf3 (diff) | |
| download | stable-diffusion-webui-gfx803-3b0127e698a2eeb913437bce0b25b478fb06ff11.tar.gz stable-diffusion-webui-gfx803-3b0127e698a2eeb913437bce0b25b478fb06ff11.tar.bz2 stable-diffusion-webui-gfx803-3b0127e698a2eeb913437bce0b25b478fb06ff11.zip | |
Merge branch 'master' of https://github.com/AUTOMATIC1111/stable-diffusion-webui into TI_optimizations
Diffstat (limited to 'modules/shared.py')
| -rw-r--r-- | modules/shared.py | 33 |
1 files changed, 31 insertions, 2 deletions
diff --git a/modules/shared.py b/modules/shared.py index 4c3d0ce7..575f50d3 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -144,9 +144,38 @@ class State: self.sampling_step = 0
self.current_image_sampling_step = 0
- def get_job_timestamp(self):
- return datetime.datetime.now().strftime("%Y%m%d%H%M%S") # shouldn't this return job_timestamp?
+ def dict(self):
+ obj = {
+ "skipped": self.skipped,
+ "interrupted": self.skipped,
+ "job": self.job,
+ "job_count": self.job_count,
+ "job_no": self.job_no,
+ "sampling_step": self.sampling_step,
+ "sampling_steps": self.sampling_steps,
+ }
+
+ return obj
+
+ def begin(self):
+ self.sampling_step = 0
+ self.job_count = -1
+ self.job_no = 0
+ self.job_timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
+ self.current_latent = None
+ self.current_image = None
+ self.current_image_sampling_step = 0
+ self.skipped = False
+ self.interrupted = False
+ self.textinfo = None
+
+ devices.torch_gc()
+
+ def end(self):
+ self.job = ""
+ self.job_count = 0
+ devices.torch_gc()
state = State()
|