100 Commits

Author	SHA1	Message	Date
tsukimiya	949ee6fcc9	Fixed --debug_dataset option to work in non-Windows environments	2023-02-03 00:37:27 +09:00
hitomi	26a81d075c	add --persistent_data_loader_workers option	2023-02-01 16:02:15 +08:00
Kohya S	74eba06d13	Merge pull request #104 from space-nuko/caption-frequency-metadata ... Add tag frequency metadata	2023-01-31 20:56:15 +09:00
Kohya S	6bbb4d426e	Fix unet config in Diffusers (sample_size=64)	2023-01-29 20:43:58 +09:00
Kohya S	ed2e431950	Merge branch 'main' into caption-frequency-metadata	2023-01-29 17:50:23 +09:00
Kohya S	3fb12e41b7	Merge branch 'main' into textual_inversion	2023-01-26 17:50:20 +09:00
Kohya S	91a50ea637	Change img_ar_errors to mean because too many imgs	2023-01-24 20:17:15 +09:00
Kohya S	36dc97c841	Merge pull request #103 from space-nuko/bucketing-metadata ... Add bucketing metadata	2023-01-24 19:06:21 +09:00
Kohya S	e6bad080cb	Merge pull request #102 from space-nuko/precalculate-hashes ... Precalculate .safetensors model hashes after training	2023-01-24 19:03:45 +09:00
Kohya S	7f17237ada	Merge pull request #92 from forestsource/add_save_n_epoch_ratio ... Add save_n_epoch_ratio	2023-01-24 18:59:47 +09:00
space-nuko	2e8a3d20dd	Add tag frequency metadata	2023-01-23 17:43:03 -08:00
space-nuko	66051883fb	Add bucketing metadata	2023-01-23 17:26:58 -08:00
space-nuko	f7fbdc4b2a	Precalculate .safetensors model hashes after training	2023-01-23 17:21:04 -08:00
forestsource	5e817e4343	Add save_n_epoch_ratio	2023-01-22 03:00:28 +09:00
Kohya S	22ee0ac467	Move TE/UN loss calc to train script	2023-01-21 12:51:17 +09:00
Kohya S	17089b1287	Merge branch 'dev' of https://github.com/kohya-ss/sd-scripts into dev	2023-01-21 12:46:20 +09:00
Kohya S	7ee808d5d7	Merge pull request #79 from mgz-dev/tensorboard-improvements ... expand details in tensorboard logs	2023-01-21 12:46:13 +09:00
Kohya S	9ff26af68b	Update to add grad_ckpting etc to metadata	2023-01-21 12:36:31 +09:00
Kohya S	7dbcef745a	Merge pull request #77 from space-nuko/ss-extra-metadata ... More helpful metadata	2023-01-21 12:18:23 +09:00
Kohya S	758323532b	add save_last_n_epochs_state to train_network	2023-01-19 20:59:45 +09:00
space-nuko	da48f74e7b	Add new version model/VAE hash to training metadata	2023-01-18 23:00:16 -08:00
michaelgzhang	303c3410e2	expand details in tensorboard logs ... - Update tensorboard logging to track both unet and textencoder learning rates - Update tensorboard logging to track both current and moving average epoch loss - Clean up tensorboard log variable names for dashboard formatting	2023-01-18 13:10:13 -06:00
space-nuko	de1dde1a06	More helpful metadata ... - dataset/reg image dirs - random session ID - keep_tokens - training date - output name	2023-01-17 16:28:35 -08:00
Yuta Hayashibe	3eb8fb1875	Make not to save state when args.save_state is False	2023-01-18 01:31:38 +09:00
Yuta Hayashibe	3815b82bef	Removed --save_last_n_epochs_model	2023-01-16 21:02:27 +09:00
Yuta Hayashibe	c6e28faa57	Save state when args.save_last_n_epochs_state is designated	2023-01-15 19:43:37 +09:00
Yuta Hayashibe	a888223869	Fix a bug	2023-01-15 18:02:17 +09:00
Yuta Hayashibe	d30ea7966d	Updated help	2023-01-15 18:00:51 +09:00
Yuta Hayashibe	df9cb2f11c	Add --save_last_n_epochs_model and --save_last_n_epochs_state	2023-01-15 17:52:22 +09:00
Kohya S	186a2665ad	Merge branch 'main' into textual_inversion	2023-01-15 16:08:53 +09:00
Kohya S	aa40cb9345	Add train epochs and max workers option to train	2023-01-15 13:07:47 +09:00
Kohya S	61ec60a893	move convert_vae to inline, restore comments	2023-01-14 21:24:09 +09:00
Hacker 17082006	dfeadf9e52	.bin file don't need to be checked	2023-01-14 15:23:46 +07:00
Hacker 17082006	b3d3f0c8ac	Not necessary to edit load_checkpoint_with_text_encoder_conversion	2023-01-14 15:07:56 +07:00
Hacker 17082006	4fe1dd6a1c	Wrong indention	2023-01-14 14:59:29 +07:00
Hacker 17082006	95ee349e2a	Edit wrong file :<	2023-01-14 14:55:57 +07:00
Hacker 17082006	a75fd3964a	Load vae and text encoder in the same way as stable-diffusion-webui	2023-01-14 14:45:55 +07:00
Kohya S	c1b14fcdd6	initial version of TI	2023-01-12 20:47:08 +09:00
Kohya S	e4f9b2b715	Add VAE to meatada, add no_metadata option	2023-01-11 23:12:18 +09:00
space-nuko	2e4ce0fdff	Add training metadata to output LoRA model	2023-01-10 02:49:52 -08:00
Kohya S	673f9ced47	Fix '*' is not working for DreamBooth	2023-01-09 21:06:58 +09:00
Gaetano Bonofiglio	d8da85b38b	fix file not found when [ is in the filename	2023-01-09 11:40:00 +01: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	9f1d3aca24	add save_state_on_train end, fix reg imgs repeats	2023-01-07 20:20:37 +09:00
Kohya S	f56988b252	unify dataset and save functions	2023-01-05 08:10:22 +09:00
Kohya S	4c35006731	split common function from train_network to util	2023-01-03 20:22:25 +09:00
Kohya S	6b522b34c1	move code for xformers to train_util	2023-01-02 16:08:21 +09:00
Kohya S	f192338874	rename another position_ids key (supports wd v1.4)	2023-01-01 09:17:16 +09:00
bmaltais	c5cca931ab	Proposed file structure rework and required file changes	2022-12-18 15:24:46 -05:00