aboutsummaryrefslogtreecommitdiffstats
path: root/modules/interrogate.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-01-04 14:04:50 +0300
committerGitHub <noreply@github.com>2023-01-04 14:04:50 +0300
commit545ae8cb1c17a038a5bfd126546acf4b9cab334c (patch)
tree77d31fa9b40d86415c524aaef80f31123412a522 /modules/interrogate.py
parenta8ad8666cd2b5df525afe702a953095877bcecde (diff)
parentd8d206c1685d1e7027d4af82ed18d106f41d1cc4 (diff)
downloadstable-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.py4
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