Commit Graph

83 Commits

Author SHA1 Message Date
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