11 Commits

Author	SHA1	Message	Date
Augusto de la Torre	6413ada2e5	Add a test for default rating	2023-03-13 01:01:29 +01:00
Augusto de la Torre	dea8c6e862	Refactor dataset to support multiple main_prompts ... ...rather than multiple whole captions	2023-03-12 19:48:15 +01:00
Augusto de la Torre	0716c40ab6	Add support for enhanced dataset configuration ... Add support for: * flip_p.txt * cond_dropout.txt * local.yaml consolidated config (including default captions) * global.yaml consolidated config which applies recursively to subfolders * flip_p, and cond_dropout config per image * manifest.json with full image-level configuration	2023-03-08 15:02:14 +01:00
Augusto de la Torre	79bcd44bed	Add docker build action ... And also a VS Code dev container	2023-02-23 02:02:56 +01:00
Joel Holdbrooks	09d95fac58	Add test for list data resolver	2023-01-29 17:21:12 -08:00
Joel Holdbrooks	326d861a86	Push DLMA into main, pass config to resolve ... This patch * passes the configuration (`argparse.Namespace`) to the resolver, * pushes the DLMA code into the main function, * makes DLMA take a `list[ImageTrainItem]` instead of `data_root`, * makes `EveryDreamBatch` take `DLMA` instead of `data_root`, etc. * allows `data_root` to be a list. By doing these things, both `EveryDreamBatch` and DLMA can be free from data resolution logic. It also reduces the number of arguments which need to be passed down to EDB and DLMA.	2023-01-29 17:08:54 -08:00
Joel Holdbrooks	620b157e6a	Fix some name errors	2023-01-23 16:57:02 -08:00
Joel Holdbrooks	685bada570	Update tests	2023-01-23 12:15:35 -08:00
Joel Holdbrooks	4e6c5f4d00	Get rid of on_event callback	2023-01-22 23:58:25 -08:00
Joel Holdbrooks	914a51b057	Add data.resolver module for training data resolution	2023-01-22 22:46:04 -08:00
Joel Holdbrooks	f4f684a915	Add unit tests for data.image_train_item	2023-01-22 22:43:59 -08:00