Commit Graph
47 Commits
Author SHA1 Message Date
aiXander a5b7621732 cleanup optimizers class and add learning rate tracking 2024-04-18 07:15:23 -07:00
xander 223969ed84 add std info to plots, change some defaults 2024-04-17 15:52:46 +02:00
aiXander c01a99c17f add token warmup 2024-04-08 19:36:54 -07:00
aiXander cec51e15d8 big refactor final 2024-04-08 00:59:00 -07:00
aiXander 8bcced05f1 cleanup #1 2024-04-07 23:17:30 -07:00
aiXander 5fdd0c1172 loss calc cleanup 2024-04-07 18:51:15 -07:00
aiXander 129ba6855c cleanup 2024-04-07 18:13:28 -07:00
aiXander cba5273546 more cleanup 2024-04-07 17:21:06 -07:00
aiXander d9a4e8f361 inference updates 2024-04-06 03:14:52 -07:00
aiXander ad952988db inference updates 2024-04-06 00:28:12 -07:00
aiXander 820edaf20b update default config 2024-04-05 03:33:12 -07:00
aiXander 3a6aa8cec3 massive cleanup 2024-04-04 19:27:33 -07:00
mayukhdeb bf91ee3e83 return filenames and prompts 2024-04-04 02:31:43 -07:00
mayukhdeb 14bfba282a return filenames of images 2024-04-03 23:53:37 -07:00
mayukhdeb 3a55b517f2 handle json stuff 2024-04-03 23:53:37 -07:00
aiXander 8d1d4560c8 update std init 2024-04-02 22:14:05 -07:00
aiXander 9dc392a7b9 first pass at fixing sd15 bug 2024-04-02 18:53:13 -07:00
aiXander 660aa3fa16 bugfixes and updates 2024-04-02 18:29:56 -07:00
mayukhdeb 161d3ad17f wip render_images_eval 2024-04-02 11:36:02 -07:00
aiXander e277232e62 tweaks 2024-04-02 03:01:00 -07:00
aiXander cd62af4049 tweaks to validation img rendering 2024-04-01 15:50:26 -07:00
mayukhdeb 8b20f2da23 imeplement aspect ratio bucketing 2024-04-01 07:33:57 -07:00
aiXander 6007d09366 Merge branch 'bughunt' of https://github.com/edenartlab/trainer into bughunt 2024-04-01 00:56:07 -07:00
aiXander cff6fd00a4 simplify preprocessing 2024-04-01 00:56:00 -07:00
mayukhdeb abead52a2f fix import 2024-03-31 04:47:26 -07:00
mayukhdeb eaaab69d8b move seed fn 2024-03-31 04:47:16 -07:00
mayukhdeb ff0c34e9da move script 2024-03-31 01:02:33 -07:00
aiXander 67445d3963 small tweaks 2024-03-30 21:17:09 -07:00
aiXander 1728715cc2 update defaults 2024-03-30 20:56:20 -07:00
aiXander d56af4c5ed minor updates 2024-03-30 20:12:46 -07:00
aiXander e3aa07ccfb add token modulation for inference 2024-03-30 19:55:19 -07:00
aiXander 3f90825f7d big refactor and add inference code 2024-03-30 19:18:16 -07:00
aiXander 6091dd7a2c update saving hooks and optimizer code 2024-03-30 15:41:06 -07:00
aiXander 11fd00e5b0 small tweaks 2024-03-30 14:17:58 -07:00
aiXander 746279f20b simplify render 2024-03-30 02:36:50 -07:00
aiXander 8ce5088064 reduce default grad clip 2024-03-29 23:21:58 -07:00
aiXander da7dbd6000 update defaults 2024-03-29 18:20:34 -07:00
aiXander b4d6a0dda8 small changes to lora 2024-03-29 16:45:03 -07:00
aiXander cd9b6629bc update args 2024-03-29 15:09:10 -07:00
mayukhdeb 7d8b7b2c63 pass trigger_text as arg 2024-03-29 12:36:28 -07:00
mayukhdeb 364dfc8157 modularization 2024-03-27 07:11:38 -07:00
mayukhdeb 21a4c42da0 dtype selection ofload 2024-03-17 05:14:06 -07:00
mayukhdeb 130db42b77 refactor to work on actual args and not config 2024-03-16 04:38:53 -07:00
mayukhdeb c185b0f0d6 offload stuff from predict.py 2024-03-16 04:38:37 -07:00
mayukhdeb c95978088c predict: offload download and model info 2024-03-15 09:21:11 -07:00
mayukhdeb 42c31241af modify config based on concept mode 2024-03-15 07:16:43 -07:00
mayukhdeb d4916bafc9 predict: move seed fn 2024-03-15 07:08:35 -07:00