Commit Graph
54 Commits
Author SHA1 Message Date
mayukhdeb 269eeec1bc slow port to a standard config 2024-03-15 06:27:18 -07:00
aiXander 17fcb39190 reset test_train 2024-03-14 01:43:57 -07:00
mayukhdeb d7c1d6005a cleanup imports 2024-03-14 01:25:58 -07:00
mayukhdeb d6af8d91d6 fix missing imports 2024-03-14 01:23:55 -07:00
aiXander f83550998e migrate some functions 2024-03-14 01:04:08 -07:00
aiXander 6240696eaf update README 2024-03-13 18:09:22 -07:00
aiXander eee4c21bd3 update TODOs 2024-03-13 18:01:59 -07:00
aiXander 1ac809f8de make caption mode an input arg 2024-03-13 17:56:06 -07:00
aiXander f0f96b98da update readme 2024-03-13 14:02:52 -07:00
aiXander f3d2d9af2c tweaks 2024-03-13 13:21:42 -07:00
aiXander 1c082d5378 add easy gpu switching 2024-03-13 13:13:06 -07:00
aiXander b60e50a705 remove .cog folder 2024-03-13 13:09:49 -07:00
aiXander 0aadd7c523 update default offset noise value 2024-03-13 13:02:33 -07:00
aiXander 964d99c2aa add offset noise training 2024-03-13 12:58:58 -07:00
aiXander 3ae086f694 tweak prodigy lr 2024-03-13 11:43:01 -07:00
aiXander f53b256c38 activate dora 2024-03-13 10:06:38 -07:00
aiXander 59b322be7e correctly load using peft 2024-03-13 09:35:29 -07:00
aiXander 924353a23a fix lora_alpha 2024-03-13 07:48:07 -07:00
aiXander 261ad19f49 add debug training img generator 2024-03-13 07:07:40 -07:00
mayukhdeb 17b008128d potential fix for saving 2024-03-13 06:37:45 -07:00
mayukhdeb d62ef200c6 use peft and disable saving for now 2024-03-13 06:09:37 -07:00
aiXander ce13a3d5d7 merge 2024-03-13 00:43:24 -07:00
aiXander 2e4ef0aa9b set grad_acc to 1 by default 2024-03-13 00:42:47 -07:00
mayukhdeb d9f4c48781 add GPU id arg 2024-03-13 00:32:25 -07:00
mayukhdeb a9a8762bc3 set gradient_accumulation_steps to 4 2024-03-12 23:50:48 -07:00
mayukhdeb 1e358bbb94 implement gradient accumulation 2024-03-12 23:50:33 -07:00
mayukhdeb e97a1dfa3b temprarily ignore all changes in tests, trainer/ and train.py 2024-03-12 23:17:48 -07:00
mayukhdeb dfe455bc3a length 2024-03-12 23:17:48 -07:00
aiXander 95cb9662de update readme 2024-03-12 18:15:20 -07:00
aiXander a19edc0eea minor updates 2024-03-12 14:36:39 -07:00
aiXander 0516229143 added gpt4-vision captioning functionality 2024-03-12 13:32:44 -07:00
mayukhdeb 83596ba79d handle masks and vae latents 2024-03-12 06:30:16 -07:00
aiXander dca204e068 tiny rewrite 2024-03-12 02:34:35 -07:00
aiXander d687d5cbbb minor changes 2024-03-12 02:03:18 -07:00
aiXander dbb4db4549 fix resolution bug for sd15 2024-03-12 01:52:07 -07:00
aiXander 194e47349d re-use training pipe for inference 2024-03-12 01:26:04 -07:00
aiXander 9bd76df1ee Merge branch 'main' of https://github.com/edenartlab/trainer into main 2024-03-12 00:44:13 -07:00
aiXander 9a47f96288 minor changes 2024-03-12 00:44:04 -07:00
mayukhdeb 6ebb7d9a50 small fixes + load and resize mask 2024-03-11 23:15:13 -07:00
aiXander 87d656c9e4 more refactoring 2024-03-11 22:18:56 -07:00
aiXander fb5b8f56c8 more refactoring 2024-03-11 21:58:22 -07:00
aiXander 925069320d allow local dir training 2024-03-11 17:03:47 -07:00
aiXander 631b21eb3d bugfixes 2024-03-11 11:17:34 -07:00
aiXander 947959c1cd Merge branch 'main' of https://github.com/edenartlab/trainer 2024-03-11 10:56:30 -07:00
aiXander f2d72d3850 minor updates 2024-03-11 10:56:24 -07:00
mayukhdeb 40fa500cf9 WIP decoupled image/caption and latents/text tokens dataset w/ support for n tokenizers 2024-03-11 07:34:34 -07:00
Xander Steenbrugge efc4273aa7 Update README.md 2024-03-10 22:26:50 -07:00
Xander Steenbrugge 813b4c4adc Update README.md 2024-03-10 22:24:26 -07:00
Xander Steenbrugge 872be6f01e Update README.md 2024-03-10 22:23:53 -07:00
Xander Steenbrugge bb8cc0fea8 Update README.md 2024-03-10 22:15:57 -07:00