diff options
| author | evshiron <evshiron@gmail.com> | 2022-10-30 09:01:01 +0000 |
|---|---|---|
| committer | evshiron <evshiron@gmail.com> | 2022-10-30 09:01:10 +0000 |
| commit | be27fd4690b1eb6c74da1e31c9696a0f1901fbba (patch) | |
| tree | 9edc185911c1d43efefd36f35e428648ed0011d0 /modules/shared.py | |
| parent | 17a2076f72562b428052ee3fc8c43d19c03ecd1e (diff) | |
| download | stable-diffusion-webui-gfx803-be27fd4690b1eb6c74da1e31c9696a0f1901fbba.tar.gz stable-diffusion-webui-gfx803-be27fd4690b1eb6c74da1e31c9696a0f1901fbba.tar.bz2 stable-diffusion-webui-gfx803-be27fd4690b1eb6c74da1e31c9696a0f1901fbba.zip | |
fix broken progress api by previous rework
Diffstat (limited to 'modules/shared.py')
| -rw-r--r-- | modules/shared.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/shared.py b/modules/shared.py index e4f163c1..2c7d28a5 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -4,6 +4,7 @@ import json import os
import sys
from collections import OrderedDict
+import time
import gradio as gr
import tqdm
@@ -132,6 +133,7 @@ class State: current_image = None
current_image_sampling_step = 0
textinfo = None
+ time_start = None
def skip(self):
self.skipped = True
@@ -168,6 +170,7 @@ class State: self.skipped = False
self.interrupted = False
self.textinfo = None
+ self.time_start = time.time()
devices.torch_gc()
|