Repository navigation

Add training example for DreamBooth. - #554

Merged

patil-suraj merged 35 commits into

huggingface:mainfrom

Victarry:dreambooth-example

Sep 27, 2022

Commits on Sep 18, 2022
- Add training example for DreamBooth.
  Victarry
  committed
  bd4d674
Commits on Sep 22, 2022
- Fix bugs.
  Victarry
  committed
  51340f9
- Update readme and default hyperparameters.
  Victarry
  committed
  88ab347
- Reformatting code with black.
  Victarry
  committed
  5bb534b
Commits on Sep 23, 2022
- Update for multi-gpu trianing.
  Victarry
  committed
  faffe23
Commits on Sep 26, 2022
- Apply suggestions from code review
  patil-suraj
  authored
  2eeabe7
- improgve sampling
  patil-suraj
  committed
  195cd46
- fix autocast
  patil-suraj
  committed
  1acc678
- improve sampling more
  patil-suraj
  committed
  627cc49
- fix saving
  patil-suraj
  committed
  f1c3c8e
- actuallu fix saving
  patil-suraj
  committed
  509e4e3
- fix saving
  patil-suraj
  committed
  eafc000
- improve dataset
  patil-suraj
  committed
  6f99f29
- fix collate fun
  patil-suraj
  committed
  392fbf3
- fix collate_fn
  patil-suraj
  committed
  d6c88f4
- fix collate fn
  patil-suraj
  committed
  a3d604e
- fix key name
  patil-suraj
  committed
  f4a91a6
- fix dataset
  patil-suraj
  committed
  8e92d69
- fix collate fn
  patil-suraj
  committed
  ef01331
- concat batch in collate fn
  patil-suraj
  committed
  c66cf4d
- Merge branch 'main' of https://github.com/huggingface/diffusers into dreambooth-example
  patil-suraj
  committed
  2894a92
- add grad ckpt
  patil-suraj
  committed
  16ecc08
- add option for 8bit adam
  patil-suraj
  committed
  87bc752
- do two forward passes for prior preservation
  patil-suraj
  committed
  661ca46
Commits on Sep 27, 2022
- Revert "do two forward passes for prior preservation"
  patil-suraj
  committed
  ce2a3be
- add option for prior_loss_weight
  patil-suraj
  committed
  248e77d
- add option for clip grad norm
  patil-suraj
  committed
  abbb614
- add more comments
  patil-suraj
  committed
  c05b043
- update readme
  patil-suraj
  committed
  90bac83
- update readme
  patil-suraj
  committed
  89991a1
- Apply suggestions from code review
  
  patil-suraj
  and
  patrickvonplaten
  authored
  265d2b1
- add docstr for dataset
  patil-suraj
  committed
  d63fa4d
- update the saving logic
  patil-suraj
  committed
  102ad70
- Update examples/dreambooth/README.md
  patil-suraj
  authored
  7ad4316
- remove unused imports
  patil-suraj
  committed
  d72c659

Commits

Commits on Sep 18, 2022

Commits on Sep 22, 2022

Commits on Sep 23, 2022

Commits on Sep 26, 2022

Commits on Sep 27, 2022