Use item axis values in SDXL tags
This commit is contained in:
@@ -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"))
|
||||
|
||||
Reference in New Issue
Block a user