Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix typo in train_dreambooth_inpaint (#1885) 

Fix typo in train_dreambooth_inpaint.
This commit is contained in:
Pedro Cuenca 2023-01-02 11:50:58 +01:00 committed by GitHub
parent 21bbc633c4
commit f769d74b0f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/research_projects/dreambooth_inpaint/train_dreambooth_inpaint.py
View File

@ -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.set_description("Steps")
for epoch in range(first_epoch, args.num_epochs):
for epoch in range(first_epoch, args.num_train_epochs):
unet.train()
for step, batch in enumerate(train_dataloader):
# Skip steps until we reach the resumed step