wassname/peft
1
0
Fork 0
mirror of https://github.com/wassname/peft.git synced 2026-09-09 11:28:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #49 from orenwang/main

Browse Source 
fix validation_steps handling in dreambooth example
This commit is contained in:
Sourab Mangrulkar
2023-02-01 15:42:35 +05:30
committed by GitHub
parent dd518985ff 80e1b262e5
commit b4d0885203
1 changed files with 1 additions and 1 deletions
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:"