42 Commits

Author	SHA1	Message	Date
Yuta Hayashibe	5f6bf29e52	Replace print with logger if they are logs (#905) ... * Add get_my_logger() * Use logger instead of print * Fix log level * Removed line-breaks for readability * Use setup_logging() * Add rich to requirements.txt * Make simple * Use logger instead of print --------- Co-authored-by: Kohya S <52813779+kohya-ss@users.noreply.github.com>	2024-02-04 18:14:34 +09:00
CjangCjengh	d31aa143f4	fix path error	2023-12-08 00:27:32 +08:00
CjangCjengh	710e777a92	fix path error	2023-12-08 00:22:13 +08:00
CjangCjengh	d653e594c2	fix path error	2023-12-06 09:48:42 +08:00
Kohya S	d0923d6710	add caption_separator option	2023-11-19 21:44:52 +09:00
Yuta Hayashibe	b2626bc7a9	Fix a typo	2023-10-26 00:51:17 +09:00
Kohya S	17813ff5b4	remove workaround for transfomers bs>1 close #869	2023-10-11 07:40:12 +09:00
Kohya S	406511c333	add error message if model.onnx doesn't exist	2023-10-09 17:08:58 +09:00
Kohya S	07d297fdbe	Merge branch 'dev' into onnx	2023-10-09 15:13:40 +09:00
Kohya S	0d4e8b50d0	change option to append_tags, minor update	2023-10-09 15:09:54 +09:00
Isotr0py	b8b84021e5	fix a typo	2023-10-08 20:49:03 +08:00
Isotr0py	70fe7e18be	add onnx to wd14 tagger	2023-10-08 20:31:10 +08:00
alexds9	9378da3c82	Fix comment	2023-10-05 21:29:46 +03:00
alexds9	a4857fa764	Add append_captions feature to wd14 tagger ... This feature allows for appending new tags to the existing content of caption files. If the caption file for an image already exists, the tags generated from the current run are appended to the existing ones. Duplicate tags are checked and avoided.	2023-10-05 21:26:09 +03:00
Kohya S	6889ee2b85	add warning for bucket_reso_steps with SDXL	2023-08-11 19:02:36 +09:00
Kohya S	94c151aea3	refactor caching latents (flip in same npz, etc)	2023-07-15 18:28:33 +09:00
Kohya S	cc3d40ca44	support sdxl in prepare scipt	2023-07-07 21:16:41 +09:00
mio	334d07bf96	Update make_captions.py ... Append sys path for make_captions.py to load blip module in the same folder to fix the error when you don't run this script under the folder	2023-06-08 23:39:06 +08:00
mio	57bc2abf41	update parser format to match a global pattern	2023-04-26 20:11:32 +08:00
Kohya S	01df1c0cc4	don't replace underscore in emoji tags like ^_^	2023-04-19 12:42:20 +09:00
Kohya S	47d61e2c02	format by black	2023-04-17 22:00:26 +09:00
Kohya S	8f6fc8daa1	add ja comment, restore arg for backward compat	2023-04-17 21:58:55 +09:00
Linaqruf	c316c63dff	fix: bring positional args back, add recursive to blip etc	2023-04-12 05:41:28 +07:00
Linaqruf	07aa000750	feat: added 7 new functionalities including recursive	2023-04-07 16:51:43 +07:00
Robert Smieja	eb66e5ebac	Extract parser setup to helper function ... - Allows users who `import` the scripts to examine the parser programmatically	2023-03-20 00:06:47 -04:00
Linaqruf	44d4cfb453	feat: added function to load training config with .toml	2023-03-12 11:52:37 +07:00
Kohya S	ea2dfd09ef	update bucketing features	2023-02-05 21:37:46 +09:00
Kohya S	1bec2bfe07	Add cleaning duplicated tags	2023-02-03 21:05:55 +09:00
Kohya S	76f53429be	Fix existing npz skip feature	2023-02-03 21:05:14 +09:00
Kohya S	73d612ff9c	Add cleaning patterns	2023-02-03 21:04:37 +09:00
Kohya S	57d8483eaf	add GIT captioning, refactoring, DataLoader	2023-02-03 08:45:33 +09:00
Kohya S	86f4e20337	Merge branch 'dev' into main	2023-01-31 21:02:18 +09:00
unknown	c20745b6e8	fix: #53	2023-01-29 22:30:45 +09:00
breakcore2	64d5ceda71	simplify arg to --recursive	2023-01-26 01:06:33 -08:00
breakcore2	2ce9ad235c	add recursive structure merge dd tags and convert to pathlib	2023-01-26 01:01:38 -08:00
breakcore2	4735b21318	add .bmp support for wd14 tagger	2023-01-06 22:21:06 -08:00
Kohya S	da05ad6339	Fix npz file name for images with dots #12	2022-12-24 21:23:40 +09:00
Kohya S	890f6d5a9c	Fix relative path to train dir is not working.	2022-12-21 17:54:31 +09:00
Kohya S	dadea12ad2	Make a folder for seperating blip files, add seed.	2022-12-20 08:18:24 +09:00
Kohya S	7cca345345	Make BLIP working without extra downloading.	2022-12-19 21:58:41 +09:00
Kohya S	5f7693be04	Move generation script to root.	2022-12-19 21:57:22 +09:00
bmaltais	c5cca931ab	Proposed file structure rework and required file changes	2022-12-18 15:24:46 -05:00