747 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Jaret Burkett	bbb57de6ec	Speed up omnigen TE loading	2025-07-05 09:32:00 -06:00
Jaret Burkett	5906a76666	Fixed issue with flux kontext forcing generation image sizes	2025-06-29 05:38:20 -06:00
Jaret Burkett	57a81bc0db	Update base model version for kontext meta	2025-06-28 14:48:36 -06:00
Jaret Burkett	843be31138	Update readme changelog	2025-06-28 12:55:23 -06:00
Jaret Burkett	8fb01e96e4	Update sponsors	2025-06-28 10:05:17 -06:00
Jaret Burkett	01a3c8a9b1	Fix device issue	2025-06-26 19:14:25 -06:00
Jaret Burkett	4f91cb7148	Fix issue with gradient checkpointing and flux kontext	2025-06-26 19:03:12 -06:00
Jaret Burkett	446b0b6989	Remove revision for kontext	2025-06-26 16:46:58 -06:00
Jaret Burkett	60ef2f1df7	Added support for FLUX.1-Kontext-dev	2025-06-26 15:24:37 -06:00
Jaret Burkett	8d9c47316a	Work on mean flow. Minor bug fixes. Omnigen improvements	2025-06-26 13:46:20 -06:00
Jaret Burkett	84c6edca7e	Merge branch 'main' into dev	2025-06-25 14:10:25 -06:00
Jaret Burkett	24cd94929e	Fix bug that can happen with fast processing dataset	2025-06-25 14:01:08 -06:00
Jaret Burkett	19ea8ecc38	Added support for finetuning OmniGen2.	2025-06-25 13:58:16 -06:00
Jaret Burkett	18513ec866	Merged in from main	2025-06-24 10:56:54 -06:00
Jaret Burkett	5e733764aa	Update version	2025-06-24 10:37:13 -06:00
Jaret Burkett	03bc431279	Fixed an issue training lumina 2	2025-06-24 10:29:47 -06:00
Jaret Burkett	f3eb1dff42	Add a config flag to trigger fast image size db builder. Add config flag to set unconditional prompt for guidance loss	2025-06-24 08:51:29 -06:00
Jaret Burkett	ba1274d99e	Added a guidance burning loss. Modified DFE to work with new model. Bug fixes	2025-06-23 08:38:27 -06:00
Jaret Burkett	8602470952	Updated diffusion feature extractor	2025-06-19 15:36:10 -06:00
Jaret Burkett	4586eb5392	Added social links to sidebar	2025-06-17 13:25:24 -06:00
Jaret Burkett	989ebfaa11	Added a basic torch profiler that can be used in config during development to find some obvious issues.	2025-06-17 13:03:39 -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	1cc663a664	Performance optimizations for pre processing the batch	2025-06-17 07:37:41 -06:00
Jaret Burkett	11f2eee53a	Hide control images from ui image viewer	2025-06-16 07:18:43 -06:00
Jaret Burkett	1c2b7298dd	More work on mean flow loss. Moved it to an adapter. Still not functioning properly though.	2025-06-16 07:17:35 -06:00
Jaret Burkett	c0314ba325	Fixed some issues with training mean flow algo. Still testing WIP	2025-06-16 07:14:59 -06:00
Jaret Burkett	cbf04b8d53	Fixed some issues with training mean flow algo. Still testing WIP	2025-06-14 12:24:00 -06:00
Jaret Burkett	0946a66576	Merge branch 'main' into dev	2025-06-12 08:11:19 -06:00
Jaret Burkett	3f0ae99d48	Version bump	2025-06-12 08:01:26 -06:00
Jaret Burkett	fc83eb7691	WIP on mean flow loss. Still a WIP.	2025-06-12 08:00:51 -06:00
Jaret Burkett	cf11f128b9	Merge pull request #304 from hameerabbasi/fix-caption-loads ... Fix caption loading	2025-06-12 07:44:12 -06:00
Hameer Abbasi	5e86139e0a	Fix NameError.	2025-06-11 15:07:20 +02:00
Hameer Abbasi	c5d6b74fea	Fix caption loading.	2025-06-11 15:05:31 +02:00
Jaret Burkett	ba5196dd4a	Merge branch 'main' into dev	2025-06-10 10:26:11 -06:00
Jaret Burkett	ffb5fe0667	Version bump	2025-06-10 10:04:32 -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	97e101522c	Increase ema feedback amount. Normalize the dfe 4 image embeds	2025-06-10 08:01:13 -06:00
Jaret Burkett	eefa93f16e	Various code to support experiments.	2025-06-09 11:19:21 -06:00
Jaret Burkett	22cdfadab6	Added new timestep weighing strategy	2025-06-04 01:16:02 -06:00
Jaret Burkett	adc31ec77d	Small updates and bug fixes for various things	2025-06-03 20:08:35 -06:00
Jaret Burkett	82b90b902e	Double tap torch install to force blackwell compatability	2025-06-02 20:22:22 -06:00
Jaret Burkett	85f4b47e79	Fix issue with setup tools requirements	2025-06-02 06:54:35 -06:00
Jaret Burkett	e20a869dc1	Update docker install for blacwell	2025-06-01 19:06:53 -06:00
Jaret Burkett	12fa109910	Updated cuda arch list on docker build	2025-06-01 13:37:11 -06:00
Jaret Burkett	7d76165dcf	Merge branch 'main' into dev	2025-06-01 13:33:40 -06:00