1623 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Kohya S	e5bab69e3a	fix alpha mask without disk cache closes #1351, ref #1339	2024-06-02 21:11:40 +09:00
Kohaku-Blueleaf	3eb27ced52	Skip the final 1 step	2024-05-31 12:24:15 +08:00
Kohaku-Blueleaf	b2363f1021	Final implementation	2024-05-31 12:20:20 +08:00
Kohya S	0d96e10b3e	Merge pull request #1339 from kohya-ss/alpha-masked-loss ... Alpha masked loss	2024-05-27 21:41:16 +09:00
Kohya S	fc85496f7e	update docs for masked loss	2024-05-27 21:25:06 +09:00
Kohya S	2870be9b52	Merge branch 'dev' into alpha-masked-loss	2024-05-27 21:08:43 +09:00
Kohya S	71ad3c0f45	Update masked_loss_README-ja.md ... add sample images	2024-05-27 21:07:57 +09:00
Kohya S	ffce3b5098	Merge pull request #1349 from rockerBOO/patch-4 ... Update issue link	2024-05-27 21:00:46 +09:00
Kohya S	a4c3155148	add doc for mask loss	2024-05-27 20:59:40 +09:00
Kohya S	58cadf476b	Merge branch 'dev' into alpha-masked-loss	2024-05-27 20:02:32 +09:00
Dave Lage	d50c1b3c5c	Update issue link	2024-05-27 01:11:01 -04:00
Kohya S	e8cfd4ba1d	fix to work cond mask and alpha mask	2024-05-26 22:01:37 +09:00
Kohya S	fb12b6d8e5	Merge pull request #1347 from rockerBOO/lora-plus-log-info ... Add LoRA+ LR Ratio info message to logger	2024-05-26 19:45:03 +09:00
rockerBOO	00513b9b70	Add LoRA+ LR Ratio info message to logger	2024-05-23 22:27:12 -04:00
Kohya S	da6fea3d97	simplify and update alpha mask to work with various cases	2024-05-19 21:26:18 +09:00
Kohya S	f2dd43e198	revert kwargs to explicit declaration	2024-05-19 19:23:59 +09:00
u-haru	db6752901f	画像のアルファチャンネルをlossのマスクとして使用するオプションを追加 (#1223) ... * Add alpha_mask parameter and apply masked loss * Fix type hint in trim_and_resize_if_required function * Refactor code to use keyword arguments in train_util.py * Fix alpha mask flipping logic * Fix alpha mask initialization * Fix alpha_mask transformation * Cache alpha_mask * Update alpha_masks to be on CPU * Set flipped_alpha_masks to Null if option disabled * Check if alpha_mask is None * Set alpha_mask to None if option disabled * Add description of alpha_mask option to docs	2024-05-19 19:07:25 +09:00
Kohya S	febc5c59fa	update README	2024-05-19 19:03:43 +09:00
Kohya S	4c798129b0	update README	2024-05-19 19:00:32 +09:00
Kohya S	38e4c602b1	Merge pull request #1277 from Cauldrath/negative_learning ... Allow negative learning rate	2024-05-19 18:55:50 +09:00
Kohya S	e4d9e3c843	remove dependency for omegaconf #ref 1284	2024-05-19 17:46:07 +09:00
Kohya S	de0e0b9468	Merge pull request #1284 from sdbds/fix_traincontrolnet ... Fix train controlnet	2024-05-19 17:39:15 +09:00
Kohya S	c68baae480	add --log_config option to enable/disable output training config	2024-05-19 17:21:04 +09:00
Kohya S	47187f7079	Merge pull request #1285 from ccharest93/main ... Hyperparameter tracking	2024-05-19 16:31:33 +09:00
Kohya S	e3ddd1fbbe	update README and format code	2024-05-19 16:26:10 +09:00
Kohya S	0640f017ab	Merge pull request #1322 from aria1th/patch-1 ... Accelerate: fix get_trainable_params in controlnet-llite training	2024-05-19 16:23:01 +09:00
Kohya S	2f19175dfe	update README	2024-05-19 15:38:37 +09:00
Kohya S	146edce693	support Diffusers' based SDXL LoRA key for inference	2024-05-18 11:05:04 +09:00
Kohya S	153764a687	add prompt option '--f' for filename	2024-05-15 20:21:49 +09:00
Kohya S	589c2aa025	update README	2024-05-13 21:20:37 +09:00
Kohya S	16677da0d9	fix create_network_from_weights doesn't work	2024-05-12 22:15:07 +09:00
Kohya S	a384bf2187	Merge pull request #1313 from rockerBOO/patch-3 ... Add caption_separator to output for subset	2024-05-12 21:36:56 +09:00
Kohya S	1c296f7229	Merge pull request #1312 from rockerBOO/patch-2 ... Fix caption_separator missing in subset schema	2024-05-12 21:33:12 +09:00
Kohya S	e96a5217c3	Merge pull request #1291 from frodo821/patch-1 ... removed unnecessary `torch` import on line 115	2024-05-12 21:14:50 +09:00
Kohya S	39b82f26e5	update readme	2024-05-12 20:58:45 +09:00
Kohya S	3701507874	raise original error if error is occured in checking latents	2024-05-12 20:56:56 +09:00
Kohya S	78020936d2	Merge pull request #1278 from Cauldrath/catch_latent_error_file ... Display name of error latent file	2024-05-12 20:46:25 +09:00
Kohya S	9ddb4d7a01	update readme and help message etc.	2024-05-12 17:55:08 +09:00
Kohya S	8d1b1acd33	Merge pull request #1266 from Zovjsra/feature/disable-mmap ... Add "--disable_mmap_load_safetensors" parameter	2024-05-12 17:43:44 +09:00
Kohya S	02298e3c4a	Merge pull request #1331 from kohya-ss/lora-plus ... Lora plus	2024-05-12 17:04:58 +09:00
Kohya S	44190416c6	update docs etc.	2024-05-12 17:01:20 +09:00
Kohya S	3c8193f642	revert lora+ for lora_fa	2024-05-12 17:00:51 +09:00
Kohya S	c6a437054a	Merge branch 'dev' into lora-plus	2024-05-12 16:18:57 +09:00
Kohya S	1ffc0b330a	fix typo	2024-05-12 16:18:43 +09:00
Kohya S	e01e148705	Merge branch 'dev' into lora-plus	2024-05-12 16:17:52 +09:00
Kohya S	e9f3a622f4	Merge branch 'dev' into lora-plus	2024-05-12 16:17:27 +09:00
Kohya S	7983d3db5f	Merge pull request #1319 from kohya-ss/fused-backward-pass ... Fused backward pass	2024-05-12 15:09:39 +09:00
Kohya S	bee8cee7e8	update README for fused optimizer	2024-05-12 15:08:52 +09:00
Kohya S	f3d2cf22ff	update README for fused optimizer	2024-05-12 15:03:02 +09:00
Kohya S	6dbc23cf63	Merge branch 'dev' into fused-backward-pass	2024-05-12 14:21:56 +09:00