Add clothing seed axis vocabulary
This commit is contained in:
+12
-1
@@ -9,6 +9,7 @@ SEED_AXIS_SALTS = {
|
||||
"category": 31,
|
||||
"subcategory": 37,
|
||||
"content": 41,
|
||||
"clothing": 41,
|
||||
"person": 43,
|
||||
"scene": 47,
|
||||
"pose": 53,
|
||||
@@ -20,7 +21,8 @@ SEED_AXIS_SALTS = {
|
||||
SEED_AXIS_ALIASES = {
|
||||
"category": ("category_seed", "category"),
|
||||
"subcategory": ("subcategory_seed", "subcategory"),
|
||||
"content": ("content_seed", "item_seed", "outfit_seed", "sexual_pose_seed", "content"),
|
||||
"content": ("content_seed", "item_seed", "sexual_pose_seed", "content"),
|
||||
"clothing": ("clothing_seed", "outfit_seed", "wardrobe_seed", "content_seed", "content"),
|
||||
"person": ("person_seed", "appearance_seed", "cast_seed", "person"),
|
||||
"scene": ("scene_seed", "scene"),
|
||||
"pose": ("pose_seed", "sexual_pose_seed", "pose"),
|
||||
@@ -33,6 +35,7 @@ SEED_LOCK_AXES = (
|
||||
"category",
|
||||
"subcategory",
|
||||
"content",
|
||||
"clothing",
|
||||
"person",
|
||||
"scene",
|
||||
"pose",
|
||||
@@ -46,6 +49,7 @@ SEED_REROLL_GROUPS = {
|
||||
"category": ("category",),
|
||||
"subcategory": ("subcategory",),
|
||||
"content": ("content",),
|
||||
"clothing": ("clothing",),
|
||||
"person": ("person",),
|
||||
"scene": ("scene",),
|
||||
"pose": ("pose", "role"),
|
||||
@@ -53,6 +57,8 @@ SEED_REROLL_GROUPS = {
|
||||
"expression": ("expression",),
|
||||
"composition": ("composition",),
|
||||
"content_pose": ("content", "pose", "role"),
|
||||
"content_clothing": ("content", "clothing"),
|
||||
"clothing_pose": ("clothing", "pose", "role"),
|
||||
"scene_pose": ("scene", "pose", "role"),
|
||||
}
|
||||
SEED_REROLL_AXIS_CHOICES = list(SEED_REROLL_GROUPS.keys())
|
||||
@@ -87,6 +93,8 @@ def normalize_reroll_axis(value: Any) -> str:
|
||||
normalized = _normal_key(value)
|
||||
aliases = {
|
||||
"contentpose": "content_pose",
|
||||
"contentclothing": "content_clothing",
|
||||
"clothingpose": "clothing_pose",
|
||||
"scenepose": "scene_pose",
|
||||
}
|
||||
normalized = aliases.get(normalized, normalized)
|
||||
@@ -101,6 +109,7 @@ def build_seed_config_json(
|
||||
category_seed: int = -1,
|
||||
subcategory_seed: int = -1,
|
||||
content_seed: int = -1,
|
||||
clothing_seed: int = -1,
|
||||
person_seed: int = -1,
|
||||
scene_seed: int = -1,
|
||||
pose_seed: int = -1,
|
||||
@@ -110,6 +119,7 @@ def build_seed_config_json(
|
||||
category_seed_mode: str = "auto",
|
||||
subcategory_seed_mode: str = "auto",
|
||||
content_seed_mode: str = "auto",
|
||||
clothing_seed_mode: str = "auto",
|
||||
person_seed_mode: str = "auto",
|
||||
scene_seed_mode: str = "auto",
|
||||
pose_seed_mode: str = "auto",
|
||||
@@ -134,6 +144,7 @@ def build_seed_config_json(
|
||||
"category_seed": axis_seed(category_seed, category_seed_mode),
|
||||
"subcategory_seed": axis_seed(subcategory_seed, subcategory_seed_mode),
|
||||
"content_seed": axis_seed(content_seed, content_seed_mode),
|
||||
"clothing_seed": axis_seed(clothing_seed, clothing_seed_mode),
|
||||
"person_seed": axis_seed(person_seed, person_seed_mode),
|
||||
"scene_seed": axis_seed(scene_seed, scene_seed_mode),
|
||||
"pose_seed": axis_seed(pose_seed, pose_seed_mode),
|
||||
|
||||
Reference in New Issue
Block a user