refactor interrogate/caption

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-02-01 15:40:19 -05:00
parent 654f44f66f
commit 2963ce127c
19 changed files with 239 additions and 159 deletions
+2 -2
View File
@@ -90,8 +90,8 @@ class Script(scripts.Script):
if append_interrogation != "None":
p.prompt = f"{original_prompt}, " if original_prompt else ""
if append_interrogation == "CLIP":
from modules.interrogate import legacy
p.prompt += legacy.interrogator.interrogate(p.init_images[0])
from modules.interrogate import openclip
p.prompt += openclip.interrogator.interrogate(p.init_images[0])
elif append_interrogation == "DeepBooru":
from modules.interrogate import deepbooru
p.prompt += deepbooru.model.tag(p.init_images[0])