Commit Graph
62 Commits
Author SHA1 Message Date
xander e1813c75d1 updates to avoid OOM when plotting hist 2024-07-06 22:18:47 +02:00
xander d8b3175b57 cleanup before sd3 integration 2024-06-13 13:57:20 +02:00
xander 41ddfbcca6 add resource monitoring 2024-05-29 23:33:42 +02:00
xander 8cd935251d add strip urls 2024-05-28 23:11:41 +02:00
aiXander 70a645ac79 fix bugs 2024-05-14 07:04:25 -07:00
xander b51727219e cleanup and try to fix last inference step 2024-05-14 11:42:57 +02:00
xander 0bd26033cd add conv2 to lora layers 2024-04-29 20:12:08 +02:00
xander 7d78a3d76d updates 2024-04-29 19:37:45 +02:00
xander 24fcc2e791 update defaults 2024-04-26 15:59:18 +02:00
xander 7d469d843c cleanup and make all this work also for sd15 2024-04-26 14:38:25 +02:00
xander af6c4676d5 bugfixes 2024-04-25 21:41:57 +02:00
xander 9ae9bc19d2 mask improvements 2024-04-25 21:21:14 +02:00
xander 7218c7e8c2 more tok reg updates 2024-04-24 17:51:42 +02:00
xander c60f3646fa update tok_reg loss 2024-04-24 16:43:22 +02:00
xander e1b05b1b99 update loss plots, reduce tok_reg value 2024-04-24 14:42:05 +02:00
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