Wire seed config into built-in clothing routes

2026-07-01 16:50:46 +02:00
parent 8c3f61ea6d
commit 12c5f73104
5 changed files with 75 additions and 10 deletions
@@ -78,8 +78,12 @@ def build_auto_weighted_row(
    minimal_clothing_ratio: float | None,
    standard_pose_ratio: float | None,
    seed: int,
+    seed_config: dict[str, int] | None = None,
 ) -> dict[str, Any]:
    batch_number = max(1, ((row_number - 1) // g.BATCH_SIZE) + 1)
+    clothing_rng = None
+    if seed_config is not None:
+        clothing_rng = seed_policy.axis_rng(seed_config, "clothing", seed, row_number)
    rows = g.build_rows(
        batch_number * g.BATCH_SIZE,
        start_index,
@@ -94,6 +98,7 @@ def build_auto_weighted_row(
        standard_pose_ratio,
        seed,
        g.EXPRESSION_SEED + seed,
+        clothing_rng=clothing_rng,
    )
    row = rows[row_number - 1]
    row["main_category"] = "auto_weighted"