hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-07 21:02:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add state to interrogate

Browse Source
This commit is contained in:
Vladimir Mandic
2023-01-03 11:01:04 -05:00
committed by GitHub
parent 192ddc04d6
commit d8d206c168
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/interrogate.py
View File

@@ -136,7 +136,8 @@ class InterrogateModels:
def interrogate(self, pil_image): def interrogate(self, pil_image):
res = "" res = ""
shared.state.begin()
shared.state.job = 'interrogate'
try: try:
if shared.cmd_opts.lowvram or shared.cmd_opts.medvram: if shared.cmd_opts.lowvram or shared.cmd_opts.medvram:
@@ -177,5 +178,6 @@ class InterrogateModels:
res += "<error>" res += "<error>"
self.unload() self.unload()
shared.state.end()
return res return res