Commit Graph

48 Commits

Author SHA1 Message Date
Victor Hall ed05f3aaa9 print bad image correctly this time 2023-01-17 16:42:08 -05:00
Victor Hall 4eac671544 make preloader throw if file is corrupt 2023-01-17 16:34:00 -05:00
Victor Hall 2b447f3f0c allow down to 256 2023-01-15 00:43:46 -05:00
Victor Hall ef90ee939e add lower res 2023-01-14 19:29:36 -05:00
Victor Hall c747a7291d add multiply.txt to duplicates images in a directory 2022-12-16 17:43:35 -05:00
nicolai256 3b0a7bb34a
modified the dataloader for PIL errors and added progress bar 2022-12-09 01:01:32 +01:00
Victor Hall 317af6ace1 fix indices error on colab due to new torch versions 2022-12-07 20:14:51 -05:00
Victor Hall 8f49c87bbb hotfix for issue in dataloading and mem clear 2022-11-19 14:16:06 -05:00
Victor Hall 7098971cd6 finalizing cropy jitter, aspect tweaks 2022-11-18 22:52:25 -05:00
Victor Hall b47c5c0ed3 unload old images for better sys ram use, fix up crop jitter 2022-11-16 13:52:06 -05:00
Victor Hall 02487c8db8 work on next release 2022-11-13 21:45:51 -05:00
Victor Hall d344964b4f cleaner 2022-11-10 19:10:11 -05:00
Victor Hall 7ae899f79a small bug 2022-11-10 19:09:08 -05:00
Victor Hall aaa732007a ok emergency over with multiaspect 2022-11-10 18:29:31 -05:00
Victor Hall 55ffe83d96 runpod work, lower batch size default 2022-11-09 14:51:33 -05:00
Victor Hall 06a3e48237 runpod maybe working ish 2022-11-08 23:00:54 -05:00
Victor Hall 60a444ebd1 remove unneeded log 2022-11-07 01:28:02 -05:00
Victor Hall 412347b0bf oopsie bug in logging 2022-11-07 01:26:09 -05:00
Victor Hall ca80fd0126 add option to use .txt file with image filename as caption source 2022-11-07 01:20:43 -05:00
Victor Hall 33a79c6dce prepping txt file reading for captions 2022-11-06 21:28:58 -05:00
Victor Hall e96a2101d2 Merge branch 'multiaspect' into main 2022-11-06 21:26:24 -05:00
Victor Hall 853a7500f4 some final cleanup and readme updates 2022-11-06 19:59:37 -05:00
Victor Hall aacbde8bc7 bunch of loader work and optimization stuff for aspect ratio support 2022-11-06 01:25:03 -04:00
Victor Hall b7779591b1 multi aspect ratio support 2022-11-05 11:41:48 -04:00
Victor Hall 9ccf8ee212 consistency for code check on ini 2022-11-04 16:17:17 -04:00
Victor Hall 99fd2939f3 ignore ini files in training data 2022-11-04 16:14:10 -04:00
Victor Hall 88e804b6a7 fix test_dl.py script 2022-11-03 20:30:46 -04:00
Victor Hall 89cc295515 multi-aspect image support work 2022-11-03 19:47:54 -04:00
Victor Hall 93bbd30caa cleaning up training process 2022-11-02 22:23:09 -04:00
Victor Hall 30f78de976 fix batch size hunting at start of training 2022-10-28 21:37:21 -04:00
Victor Hall cbca4d13ce fixed shuffle 2022-10-26 02:13:59 -04:00
Victor Hall 0ed7b0b007 grr bug in shuffle 2022-10-26 01:58:14 -04:00
Victor Hall f5aba7293d big readme update, cruft and dead code removed, reverting some stuff back to xavier og 2022-10-25 22:45:47 -04:00
Victor Hall ddb809d2c6 adding everydream training routines 2022-10-22 14:53:01 -04:00
Kane Wallmann e2462ba2dc Change dataloader to correctly pair training and regularization images 2022-10-01 13:56:27 +10:00
Kane Wallmann 648b87a200 Modifications to support multiple identifier training 2022-10-01 10:53:24 +10:00
David Bielejeski 5239367089 Turning off flip. 2022-09-28 16:18:09 -05:00
Matthew Krohn 589f6ea5ed changed 'demoura' to 'joepenna' for README.md 2022-09-26 18:06:04 -07:00
JoePenna 3d97f68047 ddpm changes 2022-09-25 22:23:08 -07:00
Christopher Gammage 25e5c2520e get rid of pycache stuff and make work with windows 2022-09-25 17:09:41 -07:00
Christopher Gammage 650c887c69 Update attention.py 2022-09-25 16:26:55 -07:00
JoePenna 56ee2882fb Merge branch 'main' of https://github.com/JoePenna/Dreambooth-Stable-Diffusion 2022-09-24 01:05:18 -07:00
JoePenna 5b9603f803
Merge branch 'XavierXiao:main' into main 2022-09-24 00:36:38 -07:00
JoePenna 2545d3565c Pulling in Xavier's changes 2022-09-24 00:36:14 -07:00
JoePenna ce7d1dffe7 Improvements by David Bielejeski, Niko Pueringer, Maki99 2022-09-24 00:32:50 -07:00
Xavier a22c61800d
Update util.py 2022-09-20 21:48:16 -07:00
JoePenna 99c2e27053 Tweaks for training people
A few tweaks to the repo to help training faces, though I've also used these same settings to train objects and styles.

Also adding a notebook to make it easier to run the code on vast.ai / runpod / other cloud computing platforms.
2022-09-17 16:00:31 -07:00
XavierXiao 8f22429d74 initial commit 2022-09-06 00:00:21 -07:00