403 Commits

Author	SHA1	Message	Date
Kohya S	c430cf481a	Merge pull request #428 from p1atdev/dev ... Add WandB logging support	2023-04-22 09:39:01 +09:00
tsukimiya	e746829b5f	おそらくlibgtk2がインストールされていない環境でcv2.waitKey() および cv2.destroyAllWindows() が動作しないので除外	2023-04-20 06:20:02 +09:00
Plat	a69b24a069	fix: tensorboard not working	2023-04-20 05:33:32 +09:00
Plat	8090daca40	fix: wandb not working without logging_dir	2023-04-20 05:14:28 +09:00
Plat	27ffd9fe3d	feat: support wandb logging	2023-04-20 01:41:12 +09:00
Kohya S	423e6c229c	support metadata json+.npz caching (no prepare)	2023-04-13 22:12:13 +09:00
Kohya S	a8632b7329	fix latents disk cache	2023-04-13 21:14:39 +09:00
Kohya S	2e9f7b5f91	cache latents to disk in dreambooth method	2023-04-12 23:10:39 +09:00
AI-Casanova	0d54609435	Merge branch 'kohya-ss:main' into weighted_captions	2023-04-07 14:55:40 -05:00
AI-Casanova	7527436549	Merge branch 'kohya-ss:main' into weighted_captions	2023-04-05 17:07:15 -05:00
Kohya S	541539a144	change method name, repo is private in default etc	2023-04-05 23:16:49 +09:00
Kohya S	74220bb52c	Merge pull request #348 from ddPn08/dev ... Added a function to upload to Huggingface and resume from Huggingface.	2023-04-05 21:47:36 +09:00
AI-Casanova	1892c82a60	Reinstantiate weighted captions after a necessary revert to Main	2023-04-02 19:43:34 +00:00
ddPn08	16ba1cec69	change async uploading to optional	2023-04-02 17:45:26 +09:00
ddPn08	8bfa50e283	small fix	2023-04-02 17:39:23 +09:00
ddPn08	c4a11e5a5a	fix help	2023-04-02 17:39:23 +09:00
ddPn08	3cc4939dd3	Implement huggingface upload for all scripts	2023-04-02 17:39:22 +09:00
ddPn08	b5ff4e816f	resume from huggingface repository	2023-04-02 17:39:21 +09:00
ddPn08	d42431d73a	Added feature to upload to huggingface	2023-04-02 17:39:10 +09:00
Yuta Hayashibe	9577a9f38d	Check needless num_warmup_steps	2023-04-01 20:33:20 +09:00
Kohya S	31069e1dc5	add comments about debice for clarify	2023-03-30 21:44:40 +09:00
Kohya S	6c28dfb417	Merge pull request #332 from guaneec/ddp-lowram ... Reduce peak RAM usage	2023-03-30 21:37:37 +09:00
Jakaline-dev	b0c33a4294	Merge remote-tracking branch 'upstream/main'	2023-03-30 01:35:38 +09:00
Kohya S	4f70e5dca6	fix to work with num_workers=0	2023-03-28 19:42:47 +09:00
Kohya S	238f01bc9c	fix images are used twice, update debug dataset	2023-03-27 20:48:21 +09:00
guaneec	3cdae0cbd2	Reduce peak RAM usage	2023-03-27 14:34:17 +08:00
Kohya S	14891523ce	fix seed for each dataset to make shuffling same	2023-03-26 22:17:03 +09:00
Kohya S	6732df93e2	Merge branch 'dev' into min-SNR	2023-03-26 17:10:53 +09:00
Kohya S	4f42f759ea	Merge pull request #322 from u-haru/feature/token_warmup ... タグ数を徐々に増やしながら学習するオプションの追加、persistent_workersに関する軽微なバグ修正	2023-03-26 17:05:59 +09:00
Jakaline-dev	a35d7ef227	Implement XTI	2023-03-26 05:26:10 +09:00
u-haru	a4b34a9c3c	blueprint_args_conflictは不要なため削除、shuffleが毎回行われる不具合修正	2023-03-26 03:26:55 +09:00
u-haru	5a3d564a30	print削除	2023-03-26 02:26:08 +09:00
u-haru	4dc1124f93	lora以外も対応	2023-03-26 02:19:55 +09:00
u-haru	292cdb8379	データセットにepoch、stepが通達されないバグ修正	2023-03-26 01:44:25 +09:00
u-haru	1b89b2a10e	シャッフル前にタグを切り詰めるように変更	2023-03-24 13:44:30 +09:00
u-haru	447c56bf50	typo修正、stepをglobal_stepに修正、バグ修正	2023-03-23 09:53:14 +09:00
u-haru	a9b26b73e0	implement token warmup	2023-03-23 07:37:14 +09:00
AI-Casanova	64c923230e	Min-SNR Weighting Strategy: Refactored and added to all trainers	2023-03-22 01:27:29 +00:00
AI-Casanova	795a6bd2d8	Merge branch 'kohya-ss:main' into min-SNR	2023-03-21 13:19:15 -05:00
Kohya S	7b324bcc3b	support extensions of image files with uppercases	2023-03-21 21:10:34 +09:00
Kohya S	6d9f3bc0b2	fix different reso in batch	2023-03-21 18:33:46 +09:00
Kohya S	1816ac3271	add vae_batch_size option for faster caching	2023-03-21 18:15:57 +09:00
Kohya S	cb08fa0379	fix no npz with full path	2023-03-21 15:05:25 +09:00
AI-Casanova	a265225972	Min-SNR Weighting Strategy	2023-03-20 22:51:38 +00:00
Kohya S	de95431895	support win with diffusers, fix extra args eval	2023-03-19 22:09:36 +09:00
Kohya S	48c1be34f3	Merge branch 'dev' into main	2023-03-19 21:58:41 +09:00
Kohya S	140b4fad43	remove default values from output config	2023-03-19 20:06:31 +09:00
Kohya S	1f7babd2c7	Fix lpwp to support sdv2 and clip skip	2023-03-19 11:10:17 +09:00
Kohya S	1214760cea	Merge branch 'dev' into main	2023-03-19 10:56:56 +09:00
Kohya S	64d85b2f51	fix num_processes, fix indent	2023-03-19 10:52:46 +09:00