Commit Graph

  • da4d0fe016 support attn mask for l+g/t5 Kohya S 2024-08-05 20:51:34 +09:00
  • 231df197dd Fix npz path for verification Kohya S 2024-08-05 20:26:30 +09:00
  • cdb2d9c516 Update train_network.py gesen2egee 2024-08-04 17:36:34 +08:00
  • aa850aa531 Update train_network.py gesen2egee 2024-08-04 17:34:20 +08:00
  • f6dbf7c419 Update train_network.py gesen2egee 2024-08-04 15:18:53 +08:00
  • a593e837f3 Update train_network.py gesen2egee 2024-08-04 15:17:30 +08:00
  • 3d68754def Update train_db.py gesen2egee 2024-08-04 15:15:42 +08:00
  • b9bdd10129 Update train_network.py gesen2egee 2024-08-04 15:11:26 +08:00
  • 96eb74f0cb Update train_db.py gesen2egee 2024-08-04 15:06:05 +08:00
  • 68162172eb Update train_db.py gesen2egee 2024-08-04 15:03:56 +08:00
  • 1db495127f Update train_db.py gesen2egee 2024-08-04 14:53:46 +08:00
  • 31507b9901 Remove unnecessary is_train changes and use apply_debiased_estimation to calculate validation loss. Balances the influence of different time steps on training performance (without affecting actual training results) gesen2egee 2024-08-02 13:15:21 +08:00
  • 002d75179a sample images for training Kohya S 2024-07-29 23:18:34 +09:00
  • 1a977e847a fix typos Kohya S 2024-07-27 13:51:50 +09:00
  • 41dee60383 Refactor caching mechanism for latents and text encoder outputs, etc. Kohya S 2024-07-27 13:50:05 +09:00
  • 9ca7a5b6cc instead cv2 LANCZOS4 resize to pil resize sdbds 2024-07-20 21:59:11 +08:00
  • 1f16b80e88 Revert "judge image size for using diff interpolation" sdbds 2024-07-20 21:35:24 +08:00
  • 2e67978ee2 Generate sample images without having CUDA (such as on Macs) Millie 2024-07-18 11:52:58 -07:00
  • 87526942a6 judge image size for using diff interpolation sdbds 2024-07-12 22:56:38 +08:00
  • 082f13658b reduce peak GPU memory usage before training Kohya S 2024-07-12 21:28:01 +09:00
  • b8896aad40 update README Kohya S 2024-07-11 08:01:23 +09:00
  • 6f0e235f2c Fix shift value in SD3 inference. Kohya S 2024-07-11 08:00:45 +09:00
  • 3d402927ef WIP: update new latents caching Kohya S 2024-07-09 23:15:38 +09:00
  • 9dc7997803 fix typo Kohya S 2024-07-09 20:37:00 +09:00
  • 3ea4fce5e0 load models one by one Kohya S 2024-07-08 22:04:43 +09:00
  • c9de7c4e9a WIP: new latents caching Kohya S 2024-07-08 19:48:28 +09:00
  • 50e3d62474 fix to work T5XXL with fp16 Kohya S 2024-07-08 19:46:23 +09:00
  • ea18d5ba6d Fix to work full_bf16 and full_fp16. Kohya S 2024-06-29 17:45:50 +09:00
  • 19086465e8 Fix fp16 mixed precision, model is in bf16 without full_bf16 Kohya S 2024-06-29 17:21:25 +09:00
  • 66cf435479 re-fix assertion ref #1389 Kohya S 2024-06-27 13:14:09 +09:00
  • 381598c8bb fix resolution in metadata for sd3 Kohya S 2024-06-26 21:15:02 +09:00
  • 828a581e29 fix assertion for experimental impl ref #1389 Kohya S 2024-06-26 20:43:31 +09:00
  • 8f2ba27869 support text_encoder_batch_size for caching Kohya S 2024-06-26 20:36:22 +09:00
  • 0b3e4f7ab6 show file name if error in load_image ref #1385 Kohya S 2024-06-25 20:03:09 +09:00
  • 4802e4aaec workaround for long caption ref #1382 Kohya S 2024-06-24 23:13:14 +09:00
  • 0fe4eafac9 fix to use zero for initial latent Kohya S 2024-06-24 23:12:48 +09:00
  • d53ea22b2a sd3 training Kohya S 2024-06-23 23:38:20 +09:00
  • a518e3c819 Merge branch 'dev' into sd3 Kohya S 2024-06-23 14:12:07 +09:00
  • 9dd1ee458c Merge branch 'main' into dev Kohya S 2024-06-23 14:11:51 +09:00
  • 25f961bc77 fix to work cache_latents/text_encoder_outputs Kohya S 2024-06-23 13:24:30 +09:00
  • e5268286bf add sd3 models and inference script Kohya S 2024-06-15 22:20:24 +09:00
  • 56bb81c9e6 add grad_hook after restore state closes #1344 Kohya S 2024-06-12 21:39:35 +09:00
  • 22413a5247 Merge pull request #1359 from kohya-ss/train_resume_step Kohya S 2024-06-11 19:52:03 +09:00
  • 18d7597b0b update README Kohya S 2024-06-11 19:51:30 +09:00
  • 4a441889d4 Merge branch 'dev' into train_resume_step Kohya S 2024-06-11 19:27:37 +09:00
  • 3259928ce4 Merge branch 'dev' of https://github.com/kohya-ss/sd-scripts into dev Kohya S 2024-06-09 19:26:42 +09:00
  • 1a104dc75e make forward/backward pathes same ref #1363 Kohya S 2024-06-09 19:26:36 +09:00
  • 58fb64819a set static graph flag when DDP ref #1363 Kohya S 2024-06-09 19:26:09 +09:00
  • 5bfe5e411b Merge pull request #1361 from shirayu/update/github_actions/crate-ci/typos-1.21.0 Kohya S 2024-06-06 21:23:24 +09:00
  • 4ecbac131a Bump crate-ci/typos from 1.19.0 to 1.21.0, fix typos, and updated _typos.toml (Close #1307) Yuta Hayashibe 2024-06-05 16:31:44 +09:00
  • 4dbcef429b update for corner cases Kohya S 2024-06-04 21:26:55 +09:00
  • 321e24d83b Merge pull request #1353 from KohakuBlueleaf/train_resume_step Kohya S 2024-06-04 19:30:11 +09:00
  • e5bab69e3a fix alpha mask without disk cache closes #1351, ref #1339 Kohya S 2024-06-02 21:11:40 +09:00
  • 3eb27ced52 Skip the final 1 step Kohaku-Blueleaf 2024-05-31 12:24:15 +08:00
  • b2363f1021 Final implementation Kohaku-Blueleaf 2024-05-31 12:20:20 +08:00
  • 0d96e10b3e Merge pull request #1339 from kohya-ss/alpha-masked-loss Kohya S 2024-05-27 21:41:16 +09:00
  • fc85496f7e update docs for masked loss Kohya S 2024-05-27 21:25:06 +09:00
  • 2870be9b52 Merge branch 'dev' into alpha-masked-loss Kohya S 2024-05-27 21:08:43 +09:00
  • 71ad3c0f45 Update masked_loss_README-ja.md Kohya S 2024-05-27 21:07:57 +09:00
  • ffce3b5098 Merge pull request #1349 from rockerBOO/patch-4 Kohya S 2024-05-27 21:00:46 +09:00
  • a4c3155148 add doc for mask loss Kohya S 2024-05-27 20:59:40 +09:00
  • 58cadf476b Merge branch 'dev' into alpha-masked-loss Kohya S 2024-05-27 20:02:32 +09:00
  • d50c1b3c5c Update issue link Dave Lage 2024-05-27 01:11:01 -04:00
  • e8cfd4ba1d fix to work cond mask and alpha mask Kohya S 2024-05-26 22:01:37 +09:00
  • fb12b6d8e5 Merge pull request #1347 from rockerBOO/lora-plus-log-info Kohya S 2024-05-26 19:45:03 +09:00
  • 00513b9b70 Add LoRA+ LR Ratio info message to logger rockerBOO 2024-05-23 22:27:12 -04:00
  • da6fea3d97 simplify and update alpha mask to work with various cases Kohya S 2024-05-19 21:26:18 +09:00
  • f2dd43e198 revert kwargs to explicit declaration Kohya S 2024-05-19 19:23:59 +09:00
  • db6752901f 画像のアルファチャンネルをlossのマスクとして使用するオプションを追加 (#1223) u-haru 2024-05-19 19:07:25 +09:00
  • febc5c59fa update README Kohya S 2024-05-19 19:03:43 +09:00
  • 4c798129b0 update README Kohya S 2024-05-19 19:00:32 +09:00
  • 38e4c602b1 Merge pull request #1277 from Cauldrath/negative_learning Kohya S 2024-05-19 18:55:50 +09:00
  • e4d9e3c843 remove dependency for omegaconf #ref 1284 Kohya S 2024-05-19 17:46:07 +09:00
  • de0e0b9468 Merge pull request #1284 from sdbds/fix_traincontrolnet Kohya S 2024-05-19 17:39:15 +09:00
  • c68baae480 add --log_config option to enable/disable output training config Kohya S 2024-05-19 17:21:04 +09:00
  • 47187f7079 Merge pull request #1285 from ccharest93/main Kohya S 2024-05-19 16:31:33 +09:00
  • e3ddd1fbbe update README and format code Kohya S 2024-05-19 16:26:10 +09:00
  • 0640f017ab Merge pull request #1322 from aria1th/patch-1 Kohya S 2024-05-19 16:23:01 +09:00
  • 2f19175dfe update README Kohya S 2024-05-19 15:38:37 +09:00
  • 146edce693 support Diffusers' based SDXL LoRA key for inference Kohya S 2024-05-18 11:05:04 +09:00
  • 153764a687 add prompt option '--f' for filename Kohya S 2024-05-15 20:21:49 +09:00
  • 589c2aa025 update README Kohya S 2024-05-13 21:20:37 +09:00
  • 16677da0d9 fix create_network_from_weights doesn't work Kohya S 2024-05-12 22:15:07 +09:00
  • a384bf2187 Merge pull request #1313 from rockerBOO/patch-3 Kohya S 2024-05-12 21:36:56 +09:00
  • 1c296f7229 Merge pull request #1312 from rockerBOO/patch-2 Kohya S 2024-05-12 21:33:12 +09:00
  • e96a5217c3 Merge pull request #1291 from frodo821/patch-1 Kohya S 2024-05-12 21:14:50 +09:00
  • 39b82f26e5 update readme Kohya S 2024-05-12 20:58:45 +09:00
  • 3701507874 raise original error if error is occured in checking latents Kohya S 2024-05-12 20:56:56 +09:00
  • 78020936d2 Merge pull request #1278 from Cauldrath/catch_latent_error_file Kohya S 2024-05-12 20:46:25 +09:00
  • 9ddb4d7a01 update readme and help message etc. Kohya S 2024-05-12 17:55:08 +09:00
  • 8d1b1acd33 Merge pull request #1266 from Zovjsra/feature/disable-mmap Kohya S 2024-05-12 17:43:44 +09:00
  • 02298e3c4a Merge pull request #1331 from kohya-ss/lora-plus Kohya S 2024-05-12 17:04:58 +09:00
  • 44190416c6 update docs etc. Kohya S 2024-05-12 17:01:20 +09:00
  • 3c8193f642 revert lora+ for lora_fa Kohya S 2024-05-12 17:00:51 +09:00
  • c6a437054a Merge branch 'dev' into lora-plus Kohya S 2024-05-12 16:18:57 +09:00
  • 1ffc0b330a fix typo Kohya S 2024-05-12 16:18:43 +09:00
  • e01e148705 Merge branch 'dev' into lora-plus Kohya S 2024-05-12 16:17:52 +09:00
  • e9f3a622f4 Merge branch 'dev' into lora-plus Kohya S 2024-05-12 16:17:27 +09:00
  • f33e155c5b simplyfy code by updating accelerate to 0.30.0 scheduler-free-opt Kohya S 2024-05-12 15:48:40 +09:00
  • 7983d3db5f Merge pull request #1319 from kohya-ss/fused-backward-pass Kohya S 2024-05-12 15:09:39 +09:00