storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-18 07:33:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Need to pass in image ref video.

Browse Source
This commit is contained in:
CrossProduct
2025-01-24 05:21:53 +00:00
parent 440738e535
commit 305a4a994d
4 changed files with 30 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/args.py
+13 -1
View File
@@ -316,6 +316,7 @@ class Args:
validation_prompts: List[str] = None
validation_images: List[str] = None
validation_videos: List[str] = None
validation_pose_videos: List[str] = None
validation_heights: List[int] = None
validation_widths: List[int] = None
validation_num_frames: List[int] = None
@@ -879,6 +880,14 @@ def _add_validation_arguments(parser: argparse.ArgumentParser) -> None:
default=None,
help="One or more image path(s)/URLs that is used during validation to verify that the model is learning. Multiple validation paths should be separated by the '--validation_prompt_seperator' string. These should correspond to the order of the validation prompts.",
)
parser.add_argument(
"--validation_pose_videos",
type=str,
default=None,
help="One or more pose_videos path(s)/URLs that is used during validation to verify that the model is learning. Multiple validation paths should be separated by the '--validation_prompt_seperator' string. These should correspond to the order of the validation prompts.",
)
parser.add_argument(
"--validation_videos",
type=str,
@@ -1079,6 +1088,9 @@ def _map_to_args_type(args: Dict[str, Any]) -> Args:
validation_prompts = args.validation_prompts.split(args.validation_separator) if args.validation_prompts else []
validation_images = args.validation_images.split(args.validation_separator) if args.validation_images else None
validation_videos = args.validation_videos.split(args.validation_separator) if args.validation_videos else None
validation_pose_videos = args.validation_pose_videos.split(args.validation_separator) if args.validation_pose_videos else None
stripped_validation_prompts = []
validation_heights = []
validation_widths = []
@@ -1106,7 +1118,7 @@ def _map_to_args_type(args: Dict[str, Any]) -> Args:
result_args.validation_num_frames = validation_num_frames
result_args.validation_images = validation_images
result_args.validation_videos = validation_videos
result_args.validation_pose_videos = validation_pose_videos
result_args.num_validation_videos_per_prompt = args.num_validation_videos
result_args.validation_every_n_epochs = args.validation_epochs
result_args.validation_every_n_steps = args.validation_steps