- 12 Apr, 2023 1 commit
-
-
Sayak Paul authored
* fix: norm group test for UNet3D. * fix: type-casting issue in controlnet training.
-
- 11 Apr, 2023 2 commits
-
-
Chanchana Sornsoontorn authored
*
⚙ ️chore(train_controlnet) fix typo in logger message *⚙ ️chore(models) refactor modules order; make them the same as calling order When printing the BasicTransformerBlock to stdout, I think it's crucial that the attributes order are shown in proper order. And also previously the "3. Feed Forward" comment was not making sense. It should have been close to self.ff but it's instead next to self.norm3 * correct many tests * remove bogus file * make style * correct more tests * finish tests * fix one more * make style * make unclip deterministic *⚙ ️chore(models/attention) reorganize comments in BasicTransformerBlock class --------- Co-authored-by:Patrick von Platen <patrick.v.platen@gmail.com>
-
Will Berman authored
-
- 10 Apr, 2023 1 commit
-
-
YiYi Xu authored
* add wandb team and min-snr loss * make style * apply feedbacks
-
- 06 Apr, 2023 1 commit
-
-
YiYi Xu authored
* fix the error when push_to_hub but not log validation * contronet_from_pt & controlnet_revision * add intermediate checkpointing to the guide
-
- 05 Apr, 2023 1 commit
-
-
YiYi Xu authored
* load_from_disk + checkpointing_steps * apply feedback
-
- 04 Apr, 2023 4 commits
-
-
YiYi Xu authored
use custom local datset Co-authored-by:
yiyixuxu <yixu310@gmail,com> Co-authored-by:
-
Lucain authored
use upload folder in training scripts Co-authored-by:testbot <lucainp@hf.co>
-
Patrick von Platen authored
-
Ernie Chu authored
* ensure validation image RGB not RGBA * ensure validation image RGB not RGBA --------- Co-authored-by:
-
- 30 Mar, 2023 1 commit
-
-
YiYi Xu authored
Co-authored-by:
-
- 29 Mar, 2023 1 commit
-
-
Sayak Paul authored
* improve stable unclip doc. * feat: add streaming support to controlnet flax training script. * fix: CLI arg. * fix: torch dataloader shuffle setting. * fix: dataset length. * fix: wandb config. * fix: steps_per_epoch in the training loop. * add: entry about streaming in the readme * get column names from iterable dataset + fix final logging --------- Co-authored-by:yiyixuxu <yixu310@gmail.com>
-
- 28 Mar, 2023 1 commit
-
-
YiYi Xu authored
* add train_controlnet_flax --------- Co-authored-by:
-
- 15 Mar, 2023 1 commit
-
-
Henrik Forstén authored
* Controlnet training code initial commit Works with circle dataset: https://github.com/lllyasviel/ControlNet/blob/main/docs/train.md * Script for adding a controlnet to existing model * Fix control image transform Control image should be in 0..1 range. * Add license header and remove more unused configs * controlnet training readme * Allow nonlocal model in add_controlnet.py * Formatting * Remove unused code * Code quality * Initialize controlnet in training script * Formatting * Address review comments * doc style * explicit constructor args and submodule names * hub dataset NOTE - not tested * empty prompts * add conditioning image * rename * remove instance data dir * image_transforms -> -1,1 . conditioning_image_transformers -> 0, 1 * nits * remove local rank config I think this isn't necessary in any of our training scripts * validation images * proportion_empty_prompts typo * weight copying to controlnet bug * call log validation fix * fix * gitignore wandb * fix progress bar and resume from checkpoint iteration * initial step fix * log multiple images * fix * fixes * tracker project name configurable * misc * add controlnet requirements.txt * update docs * image labels * small fixes * log validation using existing models for pipeline * fix for deepspeed saving * memory usage docs * Update examples/controlnet/train_controlnet.py Co-authored-by:
Sayak Paul <spsayakpaul@gmail.com> * Update examples/controlnet/train_controlnet.py Co-authored-by:
William Berman <WLBberman@gmail.com> Co-authored-by:
-
