Skip to content

GitLab

Unverified Commit f769d74b authored by Pedro Cuenca's avatar Pedro Cuenca Committed by GitHub
Browse files

Fix typo in train_dreambooth_inpaint (#1885) 

Fix typo in train_dreambooth_inpaint.
parent 21bbc633
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py
View file @ f769d74b
...@@ -672,7 +672,7 @@ def main(): ...@@ -672,7 +672,7 @@ def main():
progress_bar = tqdm(range(global_step, args.max_train_steps), disable=not accelerator.is_local_main_process) progress_bar = tqdm(range(global_step, args.max_train_steps), disable=not accelerator.is_local_main_process)
progress_bar.set_description("Steps") progress_bar.set_description("Steps")
for epoch in range(first_epoch, args.num_epochs): for epoch in range(first_epoch, args.num_train_epochs):
unet.train() unet.train()
for step, batch in enumerate(train_dataloader): for step, batch in enumerate(train_dataloader):
# Skip steps until we reach the resumed step # Skip steps until we reach the resumed step
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment