Restore clothing details for atlas prompts
This commit is contained in:
@@ -8519,6 +8519,61 @@ def smoke_pov_oral_position_routes() -> None:
|
||||
any(str(detail).lower() in restored_top_prompt for detail in restored_details),
|
||||
f"Krea2 POV Prompt Restore final prompt did not include sampled restored details: {restored_top_prompt}",
|
||||
)
|
||||
clothing_only_top_config = sxcp_nodes.NODE_CLASS_MAPPINGS["SxCPKrea2POVPromptRestore"]().build(
|
||||
True,
|
||||
False,
|
||||
False,
|
||||
False,
|
||||
True,
|
||||
top_variant_config,
|
||||
)[0]
|
||||
clothing_only_top_pair = pb.build_insta_of_pair(
|
||||
row_number=1,
|
||||
start_index=1,
|
||||
seed=3828,
|
||||
ethnicity="any",
|
||||
figure="random",
|
||||
no_plus_women=False,
|
||||
no_black=False,
|
||||
trigger=Trigger,
|
||||
prepend_trigger_to_prompt=True,
|
||||
options_json=_insta_options(
|
||||
softcore_camera_mode="from_camera_config",
|
||||
hardcore_camera_mode="from_camera_config",
|
||||
camera_detail="compact",
|
||||
),
|
||||
character_cast=_character_cast(pov_man=True),
|
||||
hardcore_position_config=clothing_only_top_config,
|
||||
location_config=_coworking_location_config(),
|
||||
hardcore_camera_config=_orbit_camera(
|
||||
horizontal_angle=45,
|
||||
vertical_angle=0,
|
||||
zoom=7.5,
|
||||
subject_focus="action",
|
||||
),
|
||||
)
|
||||
clothing_only_top_row = clothing_only_top_pair.get("hardcore_row") or {}
|
||||
clothing_only_top_axis = clothing_only_top_row.get("item_axis_values") or {}
|
||||
clothing_only_axes = clothing_only_top_axis.get("restored_prompt_axes") or []
|
||||
clothing_only_details = clothing_only_top_axis.get("restored_prompt_details") or []
|
||||
_expect(
|
||||
clothing_only_axes == ["clothing_detail"],
|
||||
f"Krea2 POV Prompt Restore clothing-only mode should restore clothing_detail, got {clothing_only_axes!r}",
|
||||
)
|
||||
_expect(
|
||||
clothing_only_details,
|
||||
"Krea2 POV Prompt Restore clothing-only mode should add a sampled clothing detail",
|
||||
)
|
||||
clothing_only_top_krea = krea_formatter.format_krea2_prompt("", metadata_json=_json(clothing_only_top_pair), target="hardcore")
|
||||
clothing_only_top_prompt = _expect_text(
|
||||
"pov_oral_top_view_restore_clothing_only.krea_prompt",
|
||||
clothing_only_top_krea.get("krea_prompt"),
|
||||
60,
|
||||
).lower()
|
||||
_expect(
|
||||
any(str(detail).lower() in clothing_only_top_prompt for detail in clothing_only_details),
|
||||
f"Krea2 POV Prompt Restore clothing-only final prompt did not include sampled clothing detail: {clothing_only_top_prompt}",
|
||||
)
|
||||
|
||||
sitting_variant_config = sxcp_nodes.NODE_CLASS_MAPPINGS["SxCPKrea2POVOralFilter"]().build(
|
||||
"replace",
|
||||
|
||||
Reference in New Issue
Block a user