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

feat(caption): add WD14 tagger with Booru Tags tab

Browse Source 
Add SmilingWolf's WD14/WaifuDiffusion tagger models for anime/illustration
tagging as a new "Booru Tags" tab in the Caption panel.

- Support 9 models (v2 and v3 variants) via HuggingFace
- ONNX backend chosen due to safetensors v3 variants exhibiting
  unacceptable accuracy loss
- Separate thresholds for general/character tags
- Batch processing with progress bar
- Consolidate debug env var to SD_INTERROGATE_DEBUG
This commit is contained in:
CalamitousFelicitousness
2026-01-19 03:16:11 +00:00
parent 42ff7aa961
commit db97c42320
6 changed files with 701 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/interrogate/vqa.py
+1 -1
View File
@@ -13,7 +13,7 @@ from modules.interrogate import vqa_detection
# Debug logging - function-based to avoid circular import
debug_enabled = os.environ.get('SD_VQA_DEBUG', None) is not None
debug_enabled = os.environ.get('SD_INTERROGATE_DEBUG', None) is not None
def debug(*args, **kwargs):
if debug_enabled: