Commit Graph

83 Commits

Author SHA1 Message Date
breakcore2
29c9008e07 Merge branch 'kohya-ss:main' into main 2023-01-13 23:04:37 -08:00
Kohya S
bf691aef69 Update README.md
Add updates.
2023-01-12 23:21:21 +09:00
Kohya S
807bdf9cc9 Merge pull request #62 from kohya-ss/dev
Add training metadata to saved models. Thanks to space-nuko!
2023-01-12 21:55:50 +09:00
Kohya S
eba142ccb2 do not save metadata in .pt/.ckpt 2023-01-12 21:52:55 +09:00
Kohya S
9fd91d26a3 Store metadata to .ckpt as value of state dict 2023-01-12 10:54:21 +09:00
Kohya S
9622082eb8 Print metadata for additional network 2023-01-11 23:12:35 +09:00
Kohya S
e4f9b2b715 Add VAE to meatada, add no_metadata option 2023-01-11 23:12:18 +09:00
Kohya S
895a599d34 Merge pull request #54 from space-nuko/add-training-metadata
Add training metadata to saved models
2023-01-11 21:12:48 +09:00
space-nuko
de37fd9906 Fix metadata loading 2023-01-10 02:56:35 -08:00
space-nuko
0c4423d9dc Add epoch number to metadata 2023-01-10 02:50:04 -08:00
space-nuko
2e4ce0fdff Add training metadata to output LoRA model 2023-01-10 02:49:52 -08:00
Kohya S
f981dfd38a Add credits 2023-01-10 17:43:35 +09:00
Kohya S
a84ca297bd Merge pull request #52 from kohya-ss/dev
Fix the issue when folder/directory name contains ``[``
2023-01-09 21:08:34 +09:00
Kohya S
673f9ced47 Fix '*' is not working for DreamBooth 2023-01-09 21:06:58 +09:00
Kohya S
c5aae65003 Merge pull request #51 from Kidel/main
fix file not found when `[` is in the filename
2023-01-09 21:03:07 +09:00
Gaetano Bonofiglio
d8da85b38b fix file not found when [ is in the filename 2023-01-09 11:40:00 +01:00
Kohya S
c4bc435bc4 Update README 2023-01-09 15:00:20 +09:00
Kohya S
4a7b814700 Merge pull request #49 from kohya-ss/refactoring_training
Refactoring training scripts
2023-01-09 14:53:02 +09:00
Kohya S
223640e1ae Add updates. 2023-01-09 14:49:56 +09:00
Kohya S
fbaf373c8a fix gradient accum not used for lr schduler 2023-01-09 13:13:37 +09:00
Kohya S
6b62c44022 fix errors in fine tuning 2023-01-08 21:40:40 +09:00
Kohya S
1945fa186d Show error if caption isn't UTF-8, add bmp support 2023-01-08 18:50:52 +09:00
Kohya S
82e585cf01 Fix full_fp16 and clip_skip==2 is not working 2023-01-08 18:49:34 +09:00
Kohya S
80af4c0c42 Set dtype if text encoder is not trained at all 2023-01-07 21:43:27 +09:00
Kohya S
9f1d3aca24 add save_state_on_train end, fix reg imgs repeats 2023-01-07 20:20:37 +09:00
Kohya S
2efced0a9a fix training starts with debug_dataset 2023-01-07 20:19:25 +09:00
Kohya S
40d1bf3809 Merge branch 'main' into refactoring_training 2023-01-07 18:08:21 +09:00
breakcore2
4735b21318 add .bmp support for wd14 tagger 2023-01-06 22:21:06 -08:00
Kohya S
fac1813ac0 Merge branch 'main' of https://github.com/kohya-ss/sd-scripts 2023-01-07 12:29:07 +09:00
Kohya S
cbfe8126d6 Update readme for error: fp16 ... requires a GPU 2023-01-07 12:29:03 +09:00
Kohya S
54928fac7b Merge pull request #43 from kohya-ss/dev
Approximate difference of two models with LoRA, support multiple modules in generating
2023-01-06 21:38:56 +09:00
Kohya S
39a0293800 Merge branch 'main' into dev 2023-01-06 21:36:19 +09:00
Kohya S
4dd22f4dc8 add script to approximate diff of two models 2023-01-06 21:36:01 +09:00
Kohya S
1b222dbf9b erase using of deleted property 2023-01-06 17:13:56 +09:00
Kohya S
d62725b644 Update README.md
update link to dreambooth doc
2023-01-05 21:35:47 +09:00
Kohya S
dcd101b3d5 Create train_db_README-ja.md
note.comからコピーして修正した
2023-01-05 21:31:41 +09:00
Kohya S
f56988b252 unify dataset and save functions 2023-01-05 08:10:22 +09:00
Kohya S
6d10233a53 Support multiple additional networks 2023-01-04 08:32:22 +09:00
Kohya S
4c35006731 split common function from train_network to util 2023-01-03 20:22:25 +09:00
Kohya S
e31177adf3 Merge branch 'refactoring_training' of https://github.com/kohya-ss/sd-scripts into refactoring_training 2023-01-02 16:14:45 +09:00
Kohya S
6b522b34c1 move code for xformers to train_util 2023-01-02 16:08:21 +09:00
Kohya S
305bda2928 Merge pull request #31 from shirayu/add_save_last_n_epochs
Add --save_last_n_epochs option
2023-01-02 08:46:07 +09:00
Yuta Hayashibe
85d8b49129 Fix calculation for the old epoch 2023-01-01 23:36:20 +09:00
Yuta Hayashibe
61a61c51ee Add --save_last_n_epochs option 2023-01-01 21:46:38 +09:00
Kohya S
bda0e8333c Merge pull request #29 from kohya-ss/dev
rename another position_ids key (supports wd v1.4)
v0.2.0
2023-01-01 09:23:42 +09:00
Kohya S
f192338874 rename another position_ids key (supports wd v1.4) 2023-01-01 09:17:16 +09:00
Kohya S
885fd9ec90 Merge pull request #25 from shirayu/fix_typos
Fix typos
2022-12-31 19:51:51 +09:00
Yuta Hayashibe
0f20453997 Fix typos 2022-12-31 14:39:36 +09:00
Kohya S
8215f12c45 Merge pull request #22 from kohya-ss/dev
set default dataset_repeats to 1, resolution check
2022-12-29 20:13:12 +09:00
Kohya S
64de791f2c set default dataset_repeats to 1, resolution check 2022-12-29 20:12:09 +09:00