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

Update train_dreambooth.py

This commit is contained in:
natkaha 2023-06-08 08:27:06 +04:00 committed by GitHub
parent 2933b6b20b
commit b320b59ff6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/dreambooth/train_dreambooth.py
View File

@ -426,7 +426,7 @@ def main(args):
gradient_accumulation_steps=args.gradient_accumulation_steps, gradient_accumulation_steps=args.gradient_accumulation_steps,
mixed_precision=args.mixed_precision, mixed_precision=args.mixed_precision,
log_with="tensorboard", log_with="tensorboard",
logging_dir=logging_dir, project_dir=logging_dir,
) )
logging.basicConfig( logging.basicConfig(