wassname/peft
1
0 Fork 0
mirror of https://github.com/wassname/peft.git synced 2026-10-30 13:21:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix validation_steps handling in dreambooth example

Browse Source
This commit is contained in:
orenwang committed 2023-01-31 10:59:46 +08:00
1 parent dd518985ff
commit 80e1b262e5
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
examples/lora_dreambooth/train_dreambooth.py
+1 -1
View File
@@ -978,7 +978,7 @@ def main(args):
if (
args.validation_prompt is not None
and step + num_update_steps_per_epoch * epoch % args.validation_steps == 0
and (step + num_update_steps_per_epoch * epoch) % args.validation_steps == 0
):
logger.info(
f"Running validation... \n Generating {args.num_validation_images} images with prompt:"