Commit Graph

Select branches

Hide Pull Requests

main

old

sd3

ti_cross_attn_loss

Mono Color

• bf91ee3e83 return filenames and prompts mayukhdeb 2024-04-04 02:30:37 -07:00
• 036a899400 update gridsearch generation aiXander 2024-04-04 00:53:08 -07:00
• 386e56a83a update gridsearch generation aiXander 2024-04-04 00:52:48 -07:00
• e517bd335b Merge branch 'bughunt' of https://github.com/edenartlab/trainer into bughunt aiXander 2024-04-04 00:04:26 -07:00
• eef877e206 push updates and hyperparam script aiXander 2024-04-04 00:04:20 -07:00
• f593fdf5a7 rename file mayukhdeb 2024-04-03 23:54:09 -07:00
• 7d46a193dc enable aspect ratio bucketing mayukhdeb 2024-04-03 23:52:31 -07:00
• 14bfba282a return filenames of images mayukhdeb 2024-04-03 23:52:04 -07:00
• 3a55b517f2 handle json stuff mayukhdeb 2024-04-03 23:51:30 -07:00
• e43cb3891f add new dependencies for eval mayukhdeb 2024-04-03 23:51:10 -07:00
• da9781fd69 eval overhaul wip (missing aesthetic checkpoint) mayukhdeb 2024-04-03 23:50:23 -07:00
• a840283f64 more improvements on token embeddings aiXander 2024-04-03 15:56:30 -07:00
• 9198b2adab make sd15 work aiXander 2024-04-03 03:31:11 -07:00
• 76c445f073 push final changes aiXander 2024-04-02 23:12:45 -07:00
• 8d1d4560c8 update std init aiXander 2024-04-02 22:14:05 -07:00
• 9dc392a7b9 first pass at fixing sd15 bug aiXander 2024-04-02 18:53:13 -07:00
• 660aa3fa16 bugfixes and updates aiXander 2024-04-02 18:29:56 -07:00
• 5fb1e6c39c update token embedding plots aiXander 2024-04-02 17:23:38 -07:00
• 161d3ad17f wip render_images_eval mayukhdeb 2024-04-02 11:36:02 -07:00
• 0161bc5935 less ugly batch loading for aspect ratio bucketing mayukhdeb 2024-04-02 06:00:45 -07:00
• c73f6462d2 fix typo mayukhdeb 2024-04-02 05:44:29 -07:00
• e277232e62 tweaks aiXander 2024-04-02 03:01:00 -07:00
• d404d59952 adjust default training res aiXander 2024-04-02 02:19:01 -07:00
• ba28aedd15 push final changes aiXander 2024-04-02 02:15:02 -07:00
• f25f40a470 bugfixes, all seems working now aiXander 2024-04-02 01:33:39 -07:00
• d28214c850 stash changes before backtracking aiXander 2024-04-01 20:33:32 -07:00
• cd62af4049 tweaks to validation img rendering aiXander 2024-04-01 15:50:26 -07:00
• 8b20f2da23 imeplement aspect ratio bucketing mayukhdeb 2024-04-01 07:25:19 -07:00
• 059f7f2645 fix moved file aiXander 2024-04-01 01:05:26 -07:00
• 6007d09366 Merge branch 'bughunt' of https://github.com/edenartlab/trainer into bughunt aiXander 2024-04-01 00:56:07 -07:00
• cff6fd00a4 simplify preprocessing aiXander 2024-04-01 00:56:00 -07:00
• abead52a2f fix import mayukhdeb 2024-03-31 04:47:26 -07:00
• eaaab69d8b move seed fn mayukhdeb 2024-03-31 04:47:16 -07:00
• 671ff7b8dc comment out grad norm stuff (it was throwing errors) mayukhdeb 2024-03-31 04:47:00 -07:00
• 12ec54e069 remove unused stuff mayukhdeb 2024-03-31 04:46:21 -07:00
• cd355d6e7e fix error: cannot import name 'convert_all_state_dict_to_peft' from 'diffusers.utils' mayukhdeb 2024-03-31 04:46:07 -07:00
• ff0c34e9da move script mayukhdeb 2024-03-31 01:02:33 -07:00
• 65d759dfeb remove unused args mayukhdeb 2024-03-31 00:33:55 -07:00
• 67445d3963 small tweaks aiXander 2024-03-30 21:17:09 -07:00
• 4e97b176a1 replace TOKEN with TOK aiXander 2024-03-30 21:00:03 -07:00
• 1728715cc2 update defaults aiXander 2024-03-30 20:56:20 -07:00
• 90feb0709c add auto-gpu picking aiXander 2024-03-30 20:37:01 -07:00
• d56af4c5ed minor updates aiXander 2024-03-30 20:12:46 -07:00
• e3aa07ccfb add token modulation for inference aiXander 2024-03-30 19:55:19 -07:00
• 3f90825f7d big refactor and add inference code aiXander 2024-03-30 19:18:16 -07:00
• 6091dd7a2c update saving hooks and optimizer code aiXander 2024-03-30 15:41:06 -07:00
• 11fd00e5b0 small tweaks aiXander 2024-03-30 14:17:58 -07:00
• 4a3112c5be update default models aiXander 2024-03-30 04:59:17 -07:00
• 746279f20b simplify render aiXander 2024-03-30 02:36:50 -07:00
• e50844e79f simplify dataloader aiXander 2024-03-30 00:14:22 -07:00
• e0de792a9d sample latents from vae distribution at each training step aiXander 2024-03-29 23:50:58 -07:00
• 8ce5088064 reduce default grad clip aiXander 2024-03-29 23:21:58 -07:00
• be368b78bb add gradient tracking aiXander 2024-03-29 22:18:29 -07:00
• b1ab40b15c more config changes and cleanup aiXander 2024-03-29 21:24:10 -07:00
• da7dbd6000 update defaults aiXander 2024-03-29 18:20:34 -07:00
• b4d6a0dda8 small changes to lora aiXander 2024-03-29 16:45:03 -07:00
• cd9b6629bc update args aiXander 2024-03-29 15:09:10 -07:00
• 6b8715663d small refactoring, adjusting defaults aiXander 2024-03-29 14:35:12 -07:00
• e19713a4b8 add training config file mayukhdeb 2024-03-29 12:36:42 -07:00
• 7d8b7b2c63 pass trigger_text as arg mayukhdeb 2024-03-29 12:36:28 -07:00
• d2b30450e4 update config args mayukhdeb 2024-03-29 12:36:06 -07:00
• 975e385503 add preprocess step + mask grads mayukhdeb 2024-03-29 12:35:44 -07:00
• 0a25118461 update todos mayukhdeb 2024-03-29 12:34:44 -07:00
• 9d383569bc fix import mayukhdeb 2024-03-29 12:34:35 -07:00
• 8b340a6edc unpushed changes old aiXander 2024-03-29 10:38:52 -07:00
• c2468dd46d small fix mayukhdeb 2024-03-29 00:52:47 -07:00
• 3ac41fbe14 small cleanup+formatting mayukhdeb 2024-03-28 05:33:38 -07:00
• e7e3eecb48 verbose imports mayukhdeb 2024-03-28 03:55:54 -07:00
• 61944f27e8 cleanup imports mayukhdeb 2024-03-28 03:48:29 -07:00
• 364dfc8157 modularization mayukhdeb 2024-03-27 07:11:38 -07:00
• 99187c1f67 update todos mayukhdeb 2024-03-27 06:45:56 -07:00
• 904a5e6237 add requirements mayukhdeb 2024-03-26 10:48:39 -07:00
• 28c13c21f8 enable training without cog mayukhdeb 2024-03-26 10:48:16 -07:00
• 67574ef1b8 config: classmethod to init from filename mayukhdeb 2024-03-26 05:33:38 -07:00
• 68e61b6c1d migration to config is complete (working) mayukhdeb 2024-03-26 05:27:05 -07:00
• 9d4bc0ae2c more progress (working) mayukhdeb 2024-03-26 01:10:32 -07:00
• fac2811fc6 more progress (working) mayukhdeb 2024-03-26 00:22:55 -07:00
• fb57183706 big changes (working) mayukhdeb 2024-03-25 23:45:56 -07:00
• 63def6ade0 more progress (working) mayukhdeb 2024-03-25 23:17:56 -07:00
• 5e893857bb more progress (working) mayukhdeb 2024-03-25 07:11:22 -07:00
• 2fef34a4b9 more progress (working) mayukhdeb 2024-03-25 05:11:45 -07:00
• 3e5a6cbe35 more progress (things are working) mayukhdeb 2024-03-25 04:50:08 -07:00
• 28eff44081 more progress (thinsg working) mayukhdeb 2024-03-25 04:26:20 -07:00
• f2764cd309 remove unused args mayukhdeb 2024-03-24 23:59:43 -07:00
• ccf010cdb3 continue migration to config (things are working) mayukhdeb 2024-03-24 22:51:10 -07:00
• d72005a9c0 slow transition to useing config (everything seems to be working) mayukhdeb 2024-03-17 05:40:40 -07:00
• 8547000fdb remove part with unknown variable: model_input mayukhdeb 2024-03-17 05:17:12 -07:00
• 21a4c42da0 dtype selection ofload mayukhdeb 2024-03-17 05:14:06 -07:00
• 500e974f5e remove unused arg (things are working) mayukhdeb 2024-03-17 00:49:01 -07:00
• 4673ec5ef4 nuke args_dict (everything working) mayukhdeb 2024-03-16 06:47:55 -07:00
• 0c8c0edbfd get rid of args_dict and initiate slow transition to using config (everything is still working) mayukhdeb 2024-03-16 06:20:14 -07:00
• 59538e6995 more args in config (things are still working) mayukhdeb 2024-03-16 05:05:46 -07:00
• dc6023b2f8 add method to save as json mayukhdeb 2024-03-16 04:39:40 -07:00
• 130db42b77 refactor to work on actual args and not config mayukhdeb 2024-03-16 04:38:53 -07:00
• c185b0f0d6 offload stuff from predict.py mayukhdeb 2024-03-16 04:38:37 -07:00
• 8326386934 predict: move config to just below main fn + change function names + offload stuff into obtain_inserting_list_tokens mayukhdeb 2024-03-16 04:38:19 -07:00
• 039c93ad70 predict: remove positional args mayukhdeb 2024-03-15 09:55:52 -07:00
• c95978088c predict: offload download and model info mayukhdeb 2024-03-15 09:21:11 -07:00
• 322b3da61e merge aiXander 2024-03-15 09:12:01 -07:00
• 7da3e64b1e updates aiXander 2024-03-15 09:10:58 -07:00