447 Commits

Author	SHA1	Message	Date
Kohya S	66edc5af7b	invert condition for checking log_with	2023-04-22 18:05:19 +09:00
saltacc	dc37fd2ff6	fix no logging command line arg	2023-04-22 01:26:31 -07:00
Kohya S	884e6bff5d	fix face_crop_aug not working on finetune method, prepare upscaler	2023-04-22 10:41:36 +09:00
Kohya S	220436244c	some minor fixes	2023-04-22 09:55:04 +09:00
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
Kohya S	6a5f87d874	disable weighted captions in TI/XTI training	2023-04-08 21:45:57 +09:00
Kohya S	a876f2d3fb	format by black	2023-04-08 21:36:35 +09:00
AI-Casanova	dbab72153f	Clean up custom_train_functions.py ... Removed commented out lines from earlier bugfix.	2023-04-08 00:44:56 -05: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
Kohya S	6f6f9b537f	Merge pull request #364 from shirayu/check_needless_num_warmup_steps ... Check needless num_warmup_steps	2023-04-03 21:38:52 +09:00
Kohya S	f407f5a686	Merge pull request #352 from rockerBOO/dataset-config ... Open dataset_config json file before load	2023-04-03 21:31:55 +09:00
AI-Casanova	1892c82a60	Reinstantiate weighted captions after a necessary revert to Main	2023-04-02 19:43:34 +00:00
ddPn08	3f339cda6f	small fix	2023-04-02 23:21:17 +09: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	45381b188c	small fix	2023-04-02 17:39:20 +09:00
ddPn08	054fb3308c	use access token	2023-04-02 17:39:19 +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
u-haru	786971d443	Merge branch 'dev' into feature/stratified_lr	2023-04-01 15:08:41 +09:00
Kohya S	4627b389ff	fix device not specified in merge_lora.py	2023-04-01 09:15:57 +09:00
Kohya S	1cd07770a4	format by black	2023-04-01 09:13:47 +09:00
u-haru	1e164b6ec3	specify device when loading state_dict	2023-03-31 12:52:39 +09:00
u-haru	41ecccb2a9	Merge branch 'kohya-ss:main' into feature/stratified_lr	2023-03-31 12:47:56 +09:00
Kohya S	8cecc676cf	Fix device issue in load_file, reduce vram usage	2023-03-31 09:05:51 +09:00
u-haru	94441fa746	繰り返し回数のないディレクトリの名前表示修正	2023-03-31 02:26:54 +09:00
rockerBOO	313f3e8286	Open dataset_config json file before load	2023-03-30 12:08:04 -04:00
u-haru	b1dffe8d9a	ファイルロードができないバグ修正(Exception: device cuda is invalid)	2023-03-31 00:11:11 +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	895b0b6ca7	Fix saving issue if epoch/step not in checkpoint	2023-03-27 21:22:32 +09:00
Kohya S	238f01bc9c	fix images are used twice, update debug dataset	2023-03-27 20:48:21 +09:00
Kohya S	43a08b4061	add ja comment	2023-03-27 20:47:27 +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