diff options
author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-01-04 14:04:50 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-01-04 14:04:50 +0300 |
commit | 545ae8cb1c17a038a5bfd126546acf4b9cab334c (patch) | |
tree | 77d31fa9b40d86415c524aaef80f31123412a522 /modules/interrogate.py | |
parent | a8ad8666cd2b5df525afe702a953095877bcecde (diff) | |
parent | d8d206c1685d1e7027d4af82ed18d106f41d1cc4 (diff) | |
download | stable-diffusion-webui-gfx803-545ae8cb1c17a038a5bfd126546acf4b9cab334c.tar.gz |
Merge pull request #6264 from vladmandic/add-state-info
add missing state info
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
|