Filter incompatible SDXL route tags

2026-06-27 18:41:17 +02:00
parent 5ca5f1b858
commit cac4fe47cd
4 changed files with 123 additions and 5 deletions
@@ -42,6 +42,7 @@ class SDXLTagRouteDependencies:
    axis_value_tags: Callable[[dict[str, Any]], list[str]]
    camera_tags: Callable[..., list[str]]
    explicit_tags: Callable[[str, float], list[str]]
+    filter_incompatible_route_tags: Callable[[list[str], dict[str, Any]], list[str]]
    softcore_pair_tags: Callable[[dict[str, Any], dict[str, Any]], list[str]]


@@ -148,6 +149,7 @@ def row_core_tags_result(request: SDXLRowTagRequest, deps: SDXLTagRouteDependenc
    )
    for tag in deps.explicit_tags(combined, request.nude_weight):
        deps.add_one(tags, seen, tag)
+    tags = deps.filter_incompatible_route_tags(tags, row)
    return SDXLTagRoute(tags)


@@ -231,6 +233,7 @@ def hard_tags_result(request: SDXLPairTagRequest, deps: SDXLTagRouteDependencies
    combined = " ".join([hard_role, hard_item, hard_clothing, expression, composition])
    for tag in deps.explicit_tags(combined, request.nude_weight):
        deps.add_one(tags, seen, tag)
+    tags = deps.filter_incompatible_route_tags(tags, row)
    return SDXLTagRoute(tags)