Commit Graph

272 Commits

Author SHA1 Message Date
Jaret Burkett 26f4f02453 Add support for Z-Image-De-Turbo 2025-12-04 10:03:13 -07:00
Jaret Burkett 6c85184441 Set zit training adapter to default to v2 2025-12-02 16:46:05 -07:00
Jaret Burkett 4e62c38df5 Add support for training Z-Image Turbo with a de-distill training adapter 2025-11-28 08:08:53 -07:00
Jaret Burkett 01cf480233 Add FLUX.2 official weights 2025-11-25 08:52:19 -07:00
Jaret Burkett dadbeda197 Update test weights 2025-11-23 10:51:50 -07:00
Jaret Burkett 0b5f3475e2 Merge branch 'main' into flux2 2025-11-23 08:18:35 -07:00
Jaret Burkett 26e4b71b57 Fix issue with parsing image info for sample info on some windows machines. 2025-11-19 08:45:05 -07:00
Jaret Burkett cd607c4902 Fix a bug that can happen if you remove a gpu from your machine. 2025-11-19 08:28:48 -07:00
Jaret Burkett af8e9ea149 Add initial support for FLUX.2 2025-11-18 11:17:38 -07:00
Jaret Burkett 2e7b2d9926 Added Differential Guidance training target 2025-11-10 09:38:25 -07:00
Jaret Burkett ee206cfa18 Added blank prompt preservation 2025-10-22 14:55:13 -06:00
Jaret Burkett ca57ffc270 When having less than 3 sample images, add spacing to the grid so images are not huge 2025-10-22 13:52:38 -06:00
Jaret Burkett 76ce757e0c Added initial support for layer offloading wit Wan 2.2 14B models. 2025-10-20 14:54:30 -06:00
Jaret Burkett 645c54d617 Fixed issue that may occour if no queue is built when starting one from the table. 2025-10-15 08:48:07 -06:00
Jaret Burkett b523d58699 Added ability to clone an existing job in the ui 2025-10-14 14:13:37 -06:00
Jaret Burkett 7e34a03113 Added queing system to the UI 2025-10-14 12:00:42 -06:00
Jaret Burkett e9c4d94256 Allow for matching target resolution with control images for Qwen Image Edit 2509 2025-10-10 14:24:27 -06:00
Jaret Burkett 1bc6dee127 Change auto_memory to be layer_offloading and allow you to set the amount to unload 2025-10-10 13:12:32 -06:00
Jaret Burkett dfc85f0b51 Add Auto Memory for qwen models in the ui 2025-10-07 10:43:52 -06:00
Jaret Burkett 1ea50d8590 Add cpu info the the job page 2025-10-07 08:30:23 -06:00
Jaret Burkett 2e9de5eb50 Add ability to delete samples from the ui 2025-09-29 04:49:32 -06:00
Jaret Burkett ebadb321e3 On samples page, auto scroll to bottom on load. Added a floating button to scroll to bottom. 2025-09-29 03:56:17 -06:00
Jaret Burkett c233a80337 Reqorked visibility toggle on samples, should help when dealing with more samples 2025-09-28 14:13:10 -06:00
Jaret Burkett c20240be82 Add advanced menu on job to allow user to do things like make a job as stopped if the status ever gets hung 2025-09-28 13:43:00 -06:00
Jaret Burkett 4e207d92cd Add seed to the sample image modal 2025-09-28 12:54:34 -06:00
Jaret Burkett f0646a0a70 Reworked ui sample image modal to show more information and function a lot better. 2025-09-27 12:50:47 -06:00
Jaret Burkett 98d35f36a9 Add hidream ARA 2025-09-27 09:31:23 -06:00
Jaret Burkett 1069dee0e4 Added ui sopport for multi control samples and datasets. Added qwen image edit 5209 to the ui 2025-09-25 11:10:02 -06:00
Jaret Burkett f74475161e Add stepped loss type 2025-09-22 15:50:12 -06:00
Jaret Burkett 2120dc5936 Upgrade job to new ui trainer to fix issue with slider config showing up on old configs. 2025-09-17 13:41:48 -06:00
Jaret Burkett 218f673e3d Added support for new concept slider training script to CLI and UI 2025-09-16 10:22:34 -06:00
Jaret Burkett 0f2239ca23 Add force sample toggle to the ui 2025-08-31 16:58:27 -06:00
Jaret Burkett fd13bd73a6 Add a Download button on samples to download all the samples as a zip file 2025-08-27 09:12:46 -06:00
Jaret Burkett 5ad190b11d Improve UI for sample images when there are no samples 2025-08-27 08:10:24 -06:00
Jaret Burkett d0338b8b0b Allow dropping images directly into dataset folder without having to open the add images modal. Improve ui flow of dataset messaging. 2025-08-25 14:04:48 -06:00
Jaret Burkett 37eda7b2e2 Add a tab to the UI to show the config file for the job. Read only. 2025-08-25 13:08:40 -06:00
Jaret Burkett 24372b5e35 Add toggles to the UI to add flipped versions of the datasets, X, Y or both. 2025-08-24 13:39:04 -06:00
Jaret Burkett 823e690703 Changed auth to use the wording 'password' instead of 'token' and give information about defaults and how to change the password. 2025-08-23 09:30:44 -06:00
Jaret Burkett e1fd411665 Added support for Chroma1 official release. Will still use single file verstion instead of the diffusers version. 2025-08-23 09:06:28 -06:00
Jaret Burkett 59ff4efae5 Add support for training Qwen Image Edit in the UI 2025-08-22 10:26:46 -06:00
Jaret Burkett 38d3814be7 Added 4bit ARAs for Wan 2.2 14b models 2025-08-21 08:16:07 -06:00
Jaret Burkett d2bbe1872c Add support for fine tuning Wan 2.2 I2V 14B 2025-08-18 11:43:32 -06:00
Jaret Burkett 8ea2cf00f6 Added training to the ui. Still testing, but everything seems to be working. 2025-08-16 05:51:37 -06:00
Jaret Burkett ca7bfa414b Increase max number of samples to 40 2025-08-16 05:27:38 -06:00
Jaret Burkett 77b10d884d Add support for training with an accuracy recovery adapter with qwen image 2025-08-12 08:21:36 -06:00
Jaret Burkett bb6db3d635 Added support for caching text embeddings. This is just initial support and will probably fail for some models. Still needs to be ompimized 2025-08-07 10:27:55 -06:00
Jaret Burkett 5d8922fca2 Add ability to designate a dataset as i2v or t2v for models that support it 2025-08-06 09:29:47 -06:00
Jaret Burkett 23327d5659 Add qwen image to the ui 2025-08-04 15:48:51 -06:00
Jaret Burkett ef1d60ba34 Update wan 2.2 5b timestep distribution to weighted. 2025-07-30 10:13:22 -06:00
Jaret Burkett ca7c5c950b Add support for Wan2.2 5B 2025-07-29 05:31:54 -06:00
Jaret Burkett e55116d8c9 Added hidream low vram options 2025-07-27 18:29:46 -06:00
Jaret Burkett 99705ec8be Add support in UI for Hidream E1 2025-07-27 18:13:36 -06:00
Jaret Burkett ed8d14225f Add ability to set the quantization type for text encoders and transformer in the ui 2025-07-27 18:00:53 -06:00
Jaret Burkett 3e14a674ac Fix upload progress for datasets in the ui 2025-07-26 09:07:30 -06:00
Jaret Burkett 523c159579 Add vram flag to some models in the ui 2025-07-24 07:02:46 -06:00
Jaret Burkett 8610c6ed7f Made it easy to add control images to the samples in the UI 2025-07-17 12:00:48 -06:00
Jaret Burkett ef5149180c Switch i2v ui defaults to weighted 2025-07-12 21:30:04 -06:00
Jaret Burkett 7ab44ae0cd Fix issue with getting captions on runpod 2025-07-11 19:16:50 +00:00
Jaret Burkett 47002b067f Add path to image for datasets on the image card 2025-07-11 11:44:48 -06:00
Jaret Burkett 8537a8557f Add simple ui settings to train Wan i2v models. 2025-07-11 11:28:40 -06:00
Jaret Burkett 611969ec1f Allow control image for omnigen training and sampling 2025-07-09 13:54:55 -06:00
Jaret Burkett 60ef2f1df7 Added support for FLUX.1-Kontext-dev 2025-06-26 15:24:37 -06:00
Jaret Burkett 19ea8ecc38 Added support for finetuning OmniGen2. 2025-06-25 13:58:16 -06:00
Jaret Burkett 4586eb5392 Added social links to sidebar 2025-06-17 13:25:24 -06:00
Jaret Burkett ff617fdaea Started doing info bubble docs on the simple ui 2025-06-17 11:00:24 -06:00
Jaret Burkett 595a6f1735 Initial setup for a cron working on the ui for various tasks 2025-06-17 07:43:34 -06:00
Jaret Burkett 11f2eee53a Hide control images from ui image viewer 2025-06-16 07:18:43 -06:00
Jaret Burkett f8fb3b9c45 Added support for sdxl and sd1.5 to the ui. 2025-06-10 10:03:54 -06:00
Jaret Burkett d5c547da43 Fixed DOP typo 2025-06-10 08:44:47 -06:00
Jaret Burkett f19f7f9486 Fixed issue with wan2.1 training in ui. Name had a typo 2025-06-10 08:42:18 -06:00
Jaret Burkett 7317ed58af Adjust the ui of the sidebar 2025-06-10 08:40:51 -06:00
Jaret Burkett 517bc294fa Update support link 2025-06-10 08:28:30 -06:00
Jaret Burkett 25e150b370 Added support for Flex.2 in the UI 2025-05-07 12:41:51 -06:00
Jaret Burkett 43cb5603ad Added chroma model to the ui. Added logic to easily pull latest, use local, or use a specific version of chroma. Allow ustom name or path in the ui for custom models 2025-05-07 12:06:30 -06:00
Jaret Burkett 79c87701e7 Add hidream to the ui 2025-04-16 13:45:21 -06:00
Jaret Burkett b0d0466efd Add better error messages if name exists when saving a job 2025-04-03 11:25:25 -06:00
Jaret Burkett b94d7aafea Have error boundary if simple job cannot be displayed due to the job being advanced 2025-03-27 19:55:40 -06:00
Jaret Burkett e4526ad4a4 Updates to handle video in a dataset on ui 2025-03-26 12:15:28 -06:00
Jaret Burkett 9a902c067f Show the console log on the job overview in the ui. 2025-03-21 11:45:36 -06:00
Jaret Burkett 0bbc69c135 Fix issue where the job would hang in the ui if it failed to start 2025-03-21 09:46:41 -06:00
Jaret Burkett c8049a483d Fix issue with auth token not on build time 2025-03-20 17:27:30 -06:00
Jaret Burkett 3a6b24f4c8 Added a way to secure the UI. Plus various bug fixes and quality of life updates 2025-03-20 08:07:09 -06:00
Jaret Burkett 6cde96ae5f Adding files I forgot to stage to the last commit 2025-03-15 11:59:41 -06:00
Jaret Burkett 1be613ed06 Added advanced mode yaml editor to the ui 2025-03-15 11:58:59 -06:00
Jaret Burkett cf4216e6b8 Add support for wan training in ui 2025-03-13 18:54:27 -06:00
Jaret Burkett 7ae31c9ae9 Added LoKr to the ui 2025-03-02 08:49:01 -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 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 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 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 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