Skip to content

Commit bce55a9

Browse files
committed
more checks.
1 parent 3eccbd1 commit bce55a9

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

examples/dreambooth/train_dreambooth_lora_flux_kontext.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -740,6 +740,8 @@ def parse_args(input_args=None):
740740
assert args.caption_column is not None
741741
assert args.dataset_name is not None
742742
assert not args.train_text_encoder
743+
if args.validation_prompt is not None:
744+
assert args.validation_image is None and os.path.exists(args.validation_image)
743745

744746
return args
745747

0 commit comments

Comments
 (0)