Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

refactor interrogate/analyze/vqa code

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-02-01 11:47:20 -05:00
parent ceaf0238fa
commit 654f44f66f
14 changed files with 64 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/loopback.py
+4 -2
View File
@@ -2,7 +2,7 @@ import math
import gradio as gr
import modules.scripts as scripts
from modules import deepbooru, images, processing, shared
from modules import images, processing
from modules.processing import Processed
from modules.shared import opts, state
@@ -90,8 +90,10 @@ class Script(scripts.Script):
if append_interrogation != "None":
p.prompt = f"{original_prompt}, " if original_prompt else ""
if append_interrogation == "CLIP":
p.prompt += shared.interrogator.interrogate(p.init_images[0])
from modules.interrogate import legacy
p.prompt += legacy.interrogator.interrogate(p.init_images[0])
elif append_interrogation == "DeepBooru":
from modules.interrogate import deepbooru
p.prompt += deepbooru.model.tag(p.init_images[0])
state.job = f"loopback iteration {i+1}/{loops} batch {n+1}/{batch_count}"