65 Commits

Author	SHA1	Message	Date
Kohya S	2c5f5c324a	Fix crash TI train close #172, tag drop wo shuffle	2023-02-11 14:41:44 +09:00
Kohya S	b03721b4d9	Add todo comment	2023-02-10 17:36:38 +09:00
Kohya S	c2e1d4b71b	fix typo	2023-02-09 21:38:01 +09:00
Kohya S	3a72e6f003	add tag dropout	2023-02-09 21:35:27 +09:00
Kohya S	e42b2f7aa9	conditional caption dropout (in progress)	2023-02-07 22:28:56 +09:00
Kohya S	f9478f0d47	Merge pull request #159 from forestsource/main ... Add Conditional Dropout options	2023-02-07 21:50:26 +09:00
Kohya S	4fc9f1f8c5	Merge pull request #157 from shirayu/improve_tag_shuffle ... Always join with ", "	2023-02-07 21:47:05 +09:00
forestsource	7db98baa86	Add dropout options	2023-02-07 00:01:30 +09:00
Kohya S	2aa27b7a4b	Update downsampling for larger image in no_upscale	2023-02-06 20:52:24 +09:00
Yuta Hayashibe	5ea5fefcd2	Always join with ", "	2023-02-06 12:29:41 +09:00
Kohya S	ea2dfd09ef	update bucketing features	2023-02-05 21:37:46 +09:00
Kohya S	b1635f4bf6	Merge pull request #144 from tsukimiya/debug_dataset_linux_support ... Fixed --debug_dataset option to work in non-Windows environments	2023-02-04 18:19:04 +09:00
Kohya S	9fd7fb813d	Merge branch 'dev' into main	2023-02-04 18:16:03 +09:00
Kohya S	93134cdd15	Add tag freq for FinetuneDataset	2023-02-03 21:03:42 +09:00
Kohya S	57d8483eaf	add GIT captioning, refactoring, DataLoader	2023-02-03 08:45:33 +09:00
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