67 Commits

Author	SHA1	Message	Date
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	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	ff617fdaea	Started doing info bubble docs on the simple ui	2025-06-17 11:00:24 -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	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	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	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	ed84c19205	Moved the job action bar to a shred component	2025-02-22 12:20:14 -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