Use item axis values in SDXL tags

This commit is contained in:
2026-06-27 18:03:39 +02:00
parent 727aea6307
commit a94cb9f8f1
4 changed files with 85 additions and 2 deletions
+5
View File
@@ -39,6 +39,7 @@ class SDXLTagRouteDependencies:
character_tags_from_descriptor: Callable[[Any], list[str]]
metadata_family_tags: Callable[[dict[str, Any]], list[str]]
formatter_hint_tags: Callable[..., list[str]]
axis_value_tags: Callable[[dict[str, Any]], list[str]]
camera_tags: Callable[..., list[str]]
explicit_tags: Callable[[str, float], list[str]]
softcore_pair_tags: Callable[[dict[str, Any], dict[str, Any]], list[str]]
@@ -109,6 +110,8 @@ def row_core_tags_result(request: SDXLRowTagRequest, deps: SDXLTagRouteDependenc
deps.add_one(tags, seen, tag)
for tag in deps.formatter_hint_tags(row):
deps.add(tags, seen, tag)
for tag in deps.axis_value_tags(row):
deps.add(tags, seen, tag)
item = deps.row_value(row, "item", ("Sexual scene", "Sexual pose", "Erotic outfit", "Clothing")) or deps.clean(
row.get("custom_item")
@@ -205,6 +208,8 @@ def hard_tags_result(request: SDXLPairTagRequest, deps: SDXLTagRouteDependencies
deps.add_one(tags, seen, tag)
for tag in deps.formatter_hint_tags(row, root):
deps.add(tags, seen, tag)
for tag in deps.axis_value_tags(row):
deps.add(tags, seen, tag)
hard_scene = deps.clean(row.get("scene_text"))
hard_item = deps.clean(row.get("item"))