diff options
| author | Vladimir Mandic <mandic00@live.com> | 2023-01-03 16:01:04 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-03 16:01:04 +0000 |
| commit | d8d206c1685d1e7027d4af82ed18d106f41d1cc4 (patch) | |
| tree | ec4757880a071f3b6143e83b7cc9131f5ef6ca88 /modules/interrogate.py | |
| parent | 192ddc04d6de0d780f73aa5fbaa8c66cd4642e1c (diff) | |
| download | stable-diffusion-webui-gfx803-d8d206c1685d1e7027d4af82ed18d106f41d1cc4.tar.gz stable-diffusion-webui-gfx803-d8d206c1685d1e7027d4af82ed18d106f41d1cc4.tar.bz2 stable-diffusion-webui-gfx803-d8d206c1685d1e7027d4af82ed18d106f41d1cc4.zip | |
add state to interrogate
Diffstat (limited to 'modules/interrogate.py')
| -rw-r--r-- | modules/interrogate.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/modules/interrogate.py b/modules/interrogate.py index 6f761c5a..738d8ff7 100644 --- a/modules/interrogate.py +++ b/modules/interrogate.py @@ -136,7 +136,8 @@ class InterrogateModels: def interrogate(self, pil_image):
res = ""
-
+ shared.state.begin()
+ shared.state.job = 'interrogate'
try:
if shared.cmd_opts.lowvram or shared.cmd_opts.medvram:
@@ -177,5 +178,6 @@ class InterrogateModels: res += "<error>"
self.unload()
+ shared.state.end()
return res
|