Commit Graph

559 Commits

Author SHA1 Message Date
Jaret Burkett
f5e40dfa62 WIP on wan 2025-03-01 16:12:52 -07:00
Jaret Burkett
acc79956aa WIP create new class to add new models more easily 2025-03-01 13:49:02 -07:00
Jaret Burkett
60539c0b0f Allow using prior loss with a custom adapter 2025-03-01 08:01:14 -07:00
Jaret Burkett
dd700f70b3 Avoid loading state dict for automagic for now until I can sort out some issues 2025-02-26 17:03:14 -07:00
Jaret Burkett
d360e76661 fixed issue with dop prompt replacement 2025-02-26 13:35:18 -07:00
Jaret Burkett
6ec23ed226 Fixed issue when doing inverted masked prior with flowmatching algos 2025-02-26 12:12:32 -07:00
Jaret Burkett
f6e16e582a Added Differential Output Preservation Loss to trainer and ui 2025-02-25 20:12:36 -07:00
Jaret Burkett
259ded9602 Fixed issue with trigger word saving in ui 2025-02-24 11:04:24 -07:00
Jaret Burkett
440ba5fb3d Spawn windows in an cmd terminal. Should be working now, but not sure on my system 2025-02-24 08:54:56 -07:00
Jaret Burkett
093f14ac19 UI Bug fixes and initial windows support 2025-02-24 08:15:22 -07:00
Jaret Burkett
f0fbd8bb53 Merge pull request #256 from ostris/ui
Added AI-Toolkit UI
2025-02-23 16:10:49 -07:00
Jaret Burkett
0a981bea2b Fixed typo 2025-02-23 16:07:38 -07:00
Jaret Burkett
1d0e3a4498 Fixed some build issues for now. Added info to the readme 2025-02-23 15:59:17 -07:00
Jaret Burkett
3c7daf49f3 Add HF token to env when spawing via ui 2025-02-23 14:52:12 -07:00
Jaret Burkett
56d8d6bd81 Capture speed from the timer for the ui 2025-02-23 14:38:46 -07:00
Jaret Burkett
3e49337a58 Set step to the last step saved at when exiting 2025-02-23 13:21:22 -07:00
Jaret Burkett
60f848a877 Send more data when loading the model to the ui 2025-02-23 12:49:54 -07:00
Jaret Burkett
b366e46f1c Added more settings to the training config 2025-02-23 12:34:52 -07:00
Jaret Burkett
a280f78c69 Added checkpoint downloader 2025-02-22 16:48:15 -07:00
Jaret Burkett
6e19e7449e Fixed some issues with gpu info refreshing 2025-02-22 14:14:23 -07:00
Jaret Burkett
a6d46ad9ae Cleanup of job page 2025-02-22 13:54:06 -07:00
Jaret Burkett
f3725578dd Cleaned up dashboard 2025-02-22 13:23:26 -07:00
Jaret Burkett
ed99c3c0c8 Moved gpu to its own widget 2025-02-22 12:43:20 -07:00
Jaret Burkett
ed84c19205 Moved the job action bar to a shred component 2025-02-22 12:20:14 -07:00
Jaret Burkett
a7a9c11d9e Fixed add image dropbox 2025-02-22 11:59:21 -07:00
Jaret Burkett
f60698d0ee Fixed some bugs with ui and lock job name to prevent issues with continuing training. 2025-02-22 11:49:36 -07:00
Jaret Burkett
5f094fb17a Added controls to the jobs table 2025-02-22 10:57:53 -07:00
Jaret Burkett
a5227cba7b Switched to a universal table library 2025-02-22 09:59:17 -07:00
Jaret Burkett
77a5e01301 Added proper icon 2025-02-22 09:05:55 -07:00
Jaret Burkett
4ef5a668c0 Make left arrow browsing only hit last image max 2025-02-21 22:04:53 -07:00
Jaret Burkett
f081d14527 Preview samples full screen and use arrow keys to navigate them 2025-02-21 21:52:24 -07:00
Jaret Burkett
710c6de1c9 Samples work in ui now 2025-02-21 20:28:52 -07:00
Jaret Burkett
2b6e66e0cb Mor ui work 2025-02-21 12:40:17 -07:00
Jaret Burkett
ab641e014f Added funding github stuff 2025-02-21 17:13:36 +00:00
Jaret Burkett
ad87f72384 Start, stop, monitor jobs from ui working. 2025-02-21 09:49:28 -07:00
Jaret Burkett
d0214c0df9 Make ui more uniform 2025-02-21 06:18:27 -07:00
Jaret Burkett
adcf884c0f Built out the ui trainer plugin with db comminication 2025-02-21 05:53:35 -07:00
Jaret Burkett
f778d979b5 Saving captions is working 2025-02-20 16:17:00 -07:00
Jaret Burkett
db3ccbba33 Handle image deletion 2025-02-20 15:58:10 -07:00
Jaret Burkett
0d2be18a9b Delete datasets 2025-02-20 14:49:03 -07:00
Jaret Burkett
bbc340e545 Cleanup and add hooks 2025-02-20 13:38:58 -07:00
Jaret Burkett
33fdfd6091 Added beginning or lokr 2025-02-20 12:47:42 -07:00
Jaret Burkett
9f6030620f Dataset uploads working 2025-02-20 12:47:01 -07:00
Jaret Burkett
b5252b5028 More ui work 2025-02-20 11:19:01 -07:00
Jaret Burkett
b0d8fc220d More ui more ui 2025-02-19 20:54:02 -07:00
Jaret Burkett
cef7d9e594 Config ui section is coming along 2025-02-19 07:52:24 -07:00
Jaret Burkett
b13fcc1039 Setup a very basic ui 2025-02-18 10:57:14 -07:00
Jaret Burkett
b32d7e552b Shamelessly beg for money 2025-02-18 05:15:29 -07:00
Jaret Burkett
4af6c5cf30 Work on supporting flex.2 potential arch 2025-02-17 14:10:25 -07:00
Jaret Burkett
1f7784510d WIP Flex 2 pipeline 2025-02-16 14:54:29 -07:00