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 |
|