Skip to content

Commit f769d74

Browse files
authored
Fix typo in train_dreambooth_inpaint (huggingface#1885)
Fix typo in train_dreambooth_inpaint.
1 parent 21bbc63 commit f769d74

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -672,7 +672,7 @@ def collate_fn(examples):
672672
progress_bar = tqdm(range(global_step, args.max_train_steps), disable=not accelerator.is_local_main_process)
673673
progress_bar.set_description("Steps")
674674

675-
for epoch in range(first_epoch, args.num_epochs):
675+
for epoch in range(first_epoch, args.num_train_epochs):
676676
unet.train()
677677
for step, batch in enumerate(train_dataloader):
678678
# Skip steps until we reach the resumed step

0 commit comments

Comments
 (0)