Commit Graph

185 Commits

Author SHA1 Message Date
Joel Holdbrooks 685bada570 Update tests 2023-01-23 12:15:35 -08:00
Victor Hall 81599bb548
Merge pull request #12 from damian0815/hf_model_download
Enable download models from huggingface
2023-01-23 12:15:21 -08:00
Joel Holdbrooks d9081e8198 Use tqdm in JSONResolver, add a docstring 2023-01-23 12:11:42 -08:00
Damian Stewart 067ea506a2 check for local files before downloading from HF 2023-01-23 21:10:04 +01:00
Joel Holdbrooks 1a0b7994f4 Move target_wh calculation to ImageTrainItem 2023-01-23 12:00:42 -08:00
Joel Holdbrooks 1dfda8d6d4 Remove unused OptionalCallable alias 2023-01-23 11:28:11 -08:00
Joel Holdbrooks 646f383188 If there are no tags just return the main prompt 2023-01-23 11:27:12 -08:00
Joel Holdbrooks c1a66317cd Forgot to set prepared_train_data 2023-01-23 11:16:34 -08:00
Damian Stewart 6ee717c807 log a message on download 2023-01-23 19:45:23 +01:00
Damian Stewart 7a78ad0dfa prevent a crash if hf_repo_subfolder is unset 2023-01-23 19:38:47 +01:00
Damian Stewart 18fec68149 don't double-download data if safetensors files exist 2023-01-23 19:33:50 +01:00
Damian Stewart 4f3b4d7dda clarify code comments 2023-01-23 19:20:37 +01:00
Damian Stewart d24dd681c0 Merge remote-tracking branch 'upstream/main' into hf_model_download 2023-01-23 19:19:22 +01:00
Victor Hall d274ff2849 clarify what to resume from on colab notebook 2023-01-23 12:21:08 -05:00
Joel Holdbrooks b6c7299baf Merge branch 'main' into refactor-data-resolution 2023-01-23 08:43:23 -08:00
Joel Holdbrooks 993eabf99a Add static methods on ImageCaption for deriving captions from various sources 2023-01-23 08:22:44 -08:00
Joel Holdbrooks 316df2db7e Use data_resolver.resolve for data loading in data_loader 2023-01-23 00:15:32 -08:00
Joel Holdbrooks 0cf2cd71de Fix mistake in ImageCaption.parse 2023-01-23 00:14:50 -08:00
Joel Holdbrooks 4e6c5f4d00 Get rid of on_event callback 2023-01-22 23:58:25 -08:00
Joel Holdbrooks 9c6df69e4e Use DirectoryResolver.unzip_all 2023-01-22 23:14:16 -08:00
Joel Holdbrooks 08813eabb5 Use DirectoryResolver.recurse_data_root 2023-01-22 23:13:05 -08:00
Joel Holdbrooks aa0a2a1765 Sync recurse_data_root changes with data loader 2023-01-22 23:09:09 -08:00
Joel Holdbrooks 914a51b057 Add data.resolver module for training data resolution 2023-01-22 22:46:04 -08:00
Joel Holdbrooks 85b6aad6a9 If tags are empty, returne __main_prompt 2023-01-22 22:44:44 -08:00
Joel Holdbrooks f4f684a915 Add unit tests for data.image_train_item 2023-01-22 22:43:59 -08:00
Victor Hall 6630ea7a50 links for colab notebook 2023-01-23 00:12:03 -05:00
Victor Hall 424d386aeb Merge branch 'main' of https://github.com/victorchall/EveryDream2trainer into main 2023-01-22 21:43:10 -05:00
Victor Hall 9f5f773c33 remove saving yaml for sd1x models, unneeded 2023-01-22 21:43:03 -05:00
Victor Hall 6ed545efa7
Merge pull request #21 from nawnie/patch-1
Update Train_Colab.ipynb
2023-01-22 17:26:54 -08:00
Joel Holdbrooks a6cabe8d7d Add static methods to ImageCaption for deriving captions from various sources 2023-01-22 16:13:50 -08:00
nawnie a11ff5ef24
Update Train_Colab.ipynb
adds the chaining to colab, this was easier then erasing my fork
2023-01-22 18:13:31 -06:00
Victor Hall 18d1da0459 bug fix and multiplytxt fraction stuff 2023-01-22 18:59:59 -05:00
Victor Hall e4ed5ff063 minor colab cleanup 2023-01-22 16:41:33 -05:00
Victor Hall f58c305cf2 colab notebook work 2023-01-22 13:01:18 -05:00
Victor Hall 33936c1626 Merge branch 'main' of https://github.com/victorchall/EveryDream2trainer into main 2023-01-22 11:06:31 -05:00
Victor Hall c60c185ee0
Merge pull request #20 from nawnie/main
a colab edition capable of running on free teir
2023-01-22 08:04:45 -08:00
nawnie 5020644213 Created using Colaboratory 2023-01-22 00:54:00 -06:00
nawnie f839614b65 Created using Colaboratory 2023-01-22 00:16:30 -06:00
Victor Hall 24b00ab35b add fractional support for multiply.txt 2023-01-22 01:15:50 -05:00
nawnie e605cc59fb
Update train_colab.py 2023-01-22 00:09:25 -06:00
nawnie 4cef13a872
Add files via upload 2023-01-22 00:03:47 -06:00
nawnie debcdd2506
Update train.py
Revert
2023-01-22 00:02:35 -06:00
nawnie 1273624a3c
Add files via upload 2023-01-21 23:57:13 -06:00
Janek Mann ba5706c5d8
Fix wandb setup
Still need two have a log_writer
2023-01-21 23:26:04 +00:00
Victor Hall 36ece59660 better undersized log file 2023-01-20 16:23:56 -05:00
Victor Hall 1c2708dc63 few sanity checks and remove keyboard 2023-01-20 09:42:24 -05:00
Victor Hall e09249b699 fix yaml detection 2023-01-18 20:24:44 -05:00
Victor Hall 23faf05512 save yaml with ckpt files for easier loading 2023-01-18 13:07:05 -05:00
Victor Hall af503f413c update train.json example 2023-01-17 19:42:07 -05:00
Victor Hall 3e803a8313 deprecate ed1_mode, autodetect 2023-01-17 12:44:18 -05:00