Pierre LeMoine
|
08d4fb6e9f
|
[dreambooth] Using already created `Path` in dataset (#681)
using already created `Path` in dataset
|
2022-10-05 12:14:30 +02:00 |
|
Yuta Hayashibe
|
7e92c5bc73
|
Fix typos (#718)
* Fix typos
* Update examples/dreambooth/train_dreambooth.py
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
|
2022-10-04 15:22:14 +02:00 |
|
Suraj Patil
|
14f4af8f5b
|
[dreambooth] fix applying clip_grad_norm_ (#686)
fix applying clip grad norm
|
2022-10-03 10:54:01 +02:00 |
|
Suraj Patil
|
ac665b6484
|
[examples/dreambooth] don't pass tensor_format to scheduler. (#649)
don't pass tensor_format
|
2022-09-27 17:24:12 +02:00 |
|
Zhenhuan Liu
|
3b747de845
|
Add training example for DreamBooth. (#554)
* Add training example for DreamBooth.
* Fix bugs.
* Update readme and default hyperparameters.
* Reformatting code with black.
* Update for multi-gpu trianing.
* Apply suggestions from code review
* improgve sampling
* fix autocast
* improve sampling more
* fix saving
* actuallu fix saving
* fix saving
* improve dataset
* fix collate fun
* fix collate_fn
* fix collate fn
* fix key name
* fix dataset
* fix collate fn
* concat batch in collate fn
* add grad ckpt
* add option for 8bit adam
* do two forward passes for prior preservation
* Revert "do two forward passes for prior preservation"
This reverts commit 661ca4677e6dccc4ad596c2ee6ca4baad4159e95.
* add option for prior_loss_weight
* add option for clip grad norm
* add more comments
* update readme
* update readme
* Apply suggestions from code review
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
* add docstr for dataset
* update the saving logic
* Update examples/dreambooth/README.md
* remove unused imports
Co-authored-by: Suraj Patil <surajp815@gmail.com>
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
|
2022-09-27 15:01:18 +02:00 |