718 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
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	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	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	b6d25fcd10	Improvements to vae trainer. Adjust denoise prediction of DFE v3	2025-05-30 12:06:47 -06:00
Jaret Burkett	ffaf2f154a	Fix issue with the way chroma handled gradient checkpointing.	2025-05-28 08:41:47 -06:00
Jaret Burkett	34f4c14cd6	Work on vae trainer	2025-05-28 07:42:48 -06:00
Jaret Burkett	79bb9be92b	Fix issue with saving chroma full finetune.	2025-05-28 07:42:30 -06:00
Jaret Burkett	79499fa795	Allow fine tuning pruned versions of chroma. Allow flash attention 2 for chroma if it is installed.	2025-05-21 07:02:50 -06:00
Jaret Burkett	48e11cf843	Fallback unwrapping logic if fails	2025-05-21 03:10:33 -06:00
Jaret Burkett	7045a01375	Fixed issue saving optimizer in some instances.	2025-05-21 02:27:55 -06:00
Jaret Burkett	fca7fd6c38	Merge branch 'main' of github.com:ostris/ai-toolkit	2025-05-21 02:20:06 -06:00
Jaret Burkett	e5181d23cd	Added some experimental training techniques. Ignore for now. Still in testing.	2025-05-21 02:19:54 -06:00
Jaret Burkett	4f896c0d8a	Fixed issue where sampling fails if doing a full finetune for some models	2025-05-17 19:37:55 +00:00
Jaret Burkett	01101be196	version bump	2025-05-17 05:50:12 -06:00
Jaret Burkett	6174ba474e	Fixed issue with chroma sampling	2025-05-10 18:30:23 +00:00
Jaret Burkett	66a41e49d9	Bump version	2025-05-08 17:37:28 -06:00
Jaret Burkett	1210050ead	Reworked control generator. It is now significantly faster. Also uses better pose model with better license.	2025-05-08 14:35:55 -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	d9700bdb99	Added initial support for f-lite model	2025-05-01 11:15:18 -06:00
Jaret Burkett	5890e67a46	Various bug fixes	2025-04-29 09:30:33 -06:00
Jaret Burkett	2b4c525489	Reworked automagic optimizer and did more testing. Starting to really like it. Working well.	2025-04-28 08:01:10 -06:00
Jaret Burkett	88b3fbae37	Various experiments and minor bug fixes for edge cases	2025-04-25 13:44:38 -06:00
Jaret Burkett	8ff85ba14f	Add Flex2 training example	2025-04-22 11:59:47 -06:00
Jaret Burkett	80f73ce9c0	Update README.md	2025-04-22 10:46:51 -06:00
Jaret Burkett	9f42944056	Update Sponsors	2025-04-22 09:44:56 -06:00
Jaret Burkett	add83df5cc	Fixed issue with training hidream when batch size is larger than 1	2025-04-21 17:26:29 +00:00
Jaret Burkett	12e3095d8a	Fixed issue with saving base model version	2025-04-19 14:34:01 -06:00
Jaret Burkett	77001ee77f	Upodate model tag on loras	2025-04-19 10:41:27 -06:00
Jaret Burkett	d455e76c4f	Cleanup	2025-04-18 11:44:49 -06:00
Jaret Burkett	1628884254	Remove submodule install from docker	2025-04-18 10:41:52 -06:00