294 Commits

Author	SHA1	Message	Date
Kohya S	e45d3f8634	add merge LoRA script	2024-08-16 22:19:21 +09:00
Kohya S	3921a4efda	add t5xxl max token length, support schnell	2024-08-16 17:06:05 +09:00
Kohya S	35b6cb0cd1	update for torchvision	2024-08-15 22:07:35 +09:00
Kohya S	7db4222119	add sample image generation during training	2024-08-14 22:15:26 +09:00
Kohya S	56d7651f08	add experimental split mode for FLUX	2024-08-13 22:28:39 +09:00
kohya-ss	9711c96f96	update README	2024-08-13 21:03:17 +09:00
Kohya S	a7d5dabde3	Update readme	2024-08-12 17:09:19 +09:00
kohya-ss	4af36f9632	update to work interactive mode	2024-08-12 13:24:10 +09:00
Kohya S	9e09a69df1	update README	2024-08-12 08:19:45 +09:00
Kohya S	d25ae361d0	fix apply_t5_attn_mask to work	2024-08-11 19:07:07 +09:00
Kohya S	82314ac2e7	update readme for ai toolkit settings	2024-08-11 11:14:08 +09:00
Kohya S	8a0f12dde8	update FLUX LoRA training	2024-08-10 23:42:05 +09:00
Kohya S	36b2e6fc28	add FLUX.1 LoRA training	2024-08-09 22:56:48 +09:00
Kohya S	41dee60383	Refactor caching mechanism for latents and text encoder outputs, etc.	2024-07-27 13:50:05 +09:00
Kohya S	b8896aad40	update README	2024-07-11 08:01:23 +09:00
Kohya S	19086465e8	Fix fp16 mixed precision, model is in bf16 without full_bf16	2024-06-29 17:21:25 +09:00
Kohya S	d53ea22b2a	sd3 training	2024-06-23 23:38:20 +09:00
Kohya S	9dd1ee458c	Merge branch 'main' into dev	2024-06-23 14:11:51 +09:00
Kohya S	25f961bc77	fix to work cache_latents/text_encoder_outputs	2024-06-23 13:24:30 +09:00
Kohya S	18d7597b0b	update README	2024-06-11 19:51:30 +09:00
Kohya S	fc85496f7e	update docs for masked loss	2024-05-27 21:25:06 +09:00
Dave Lage	d50c1b3c5c	Update issue link	2024-05-27 01:11:01 -04: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	e4d9e3c843	remove dependency for omegaconf #ref 1284	2024-05-19 17:46:07 +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	e3ddd1fbbe	update README and format code	2024-05-19 16:26:10 +09:00
Kohya S	2f19175dfe	update README	2024-05-19 15:38:37 +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	39b82f26e5	update readme	2024-05-12 20:58:45 +09:00
Kohya S	9ddb4d7a01	update readme and help message etc.	2024-05-12 17:55:08 +09:00
Kohya S	44190416c6	update docs etc.	2024-05-12 17:01:20 +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	c1ba0b4356	update readme	2024-05-12 14:21:10 +09:00
Kohya S	bfb352bc43	change huber_schedule from exponential to snr	2024-04-07 21:07:52 +09:00
Kohya S	c973b29da4	update readme	2024-04-07 20:51:52 +09:00
Kohya S	dfa30790a9	update readme	2024-04-07 20:34:26 +09:00
Kohya S	d30ebb205c	update readme, add metadata for network module	2024-04-07 14:58:17 +09:00
Kohya S	089727b5ee	update readme	2024-04-07 12:42:49 +09:00
Kohya S	2258a1b753	add save/load hook to remove U-Net/TEs from state	2024-03-31 15:50:35 +09:00
Kohya S	cae5aa0a56	update wd14 tagger and doc	2024-03-30 21:48:22 +09:00
Kohya S	434dc408f9	update readme	2024-03-30 17:12:36 +09:00
Kohya S	6c08e97e1f	update readme	2024-03-26 20:48:08 +09:00
Kohya S	ab1e389347	Merge branch 'dev' into masked-loss	2024-03-26 19:39:30 +09:00
Kohya S	9c4492b58a	fix pytorch version 2.1.1 to 2.1.2	2024-03-24 23:17:25 +09:00
Kohya S	9bbb28c361	update PyTorch version and reorganize dependencies	2024-03-24 22:06:37 +09:00
Kohya S	8d5858826f	Merge branch 'dev' into masked-loss	2024-03-24 18:19:53 +09:00
Kohya S	381c44955e	update readme and typing hint	2024-03-24 11:27:18 +09:00