Commit Graph

Select branches

Hide Pull Requests

evan_dev

main

Mono Color

• 446b0b6989 Remove revision for kontext Jaret Burkett 2025-06-26 16:46:58 -06:00
• 60ef2f1df7 Added support for FLUX.1-Kontext-dev Jaret Burkett 2025-06-26 15:24:37 -06:00
• 8d9c47316a Work on mean flow. Minor bug fixes. Omnigen improvements Jaret Burkett 2025-06-26 13:46:20 -06:00
• 84c6edca7e Merge branch 'main' into dev Jaret Burkett 2025-06-25 14:10:25 -06:00
• 24cd94929e Fix bug that can happen with fast processing dataset Jaret Burkett 2025-06-25 14:01:08 -06:00
• 19ea8ecc38 Added support for finetuning OmniGen2. Jaret Burkett 2025-06-25 13:58:16 -06:00
• 18513ec866 Merged in from main Jaret Burkett 2025-06-24 10:56:54 -06:00
• 5e733764aa Update version Jaret Burkett 2025-06-24 10:37:13 -06:00
• 03bc431279 Fixed an issue training lumina 2 Jaret Burkett 2025-06-24 10:29:47 -06:00
• f3eb1dff42 Add a config flag to trigger fast image size db builder. Add config flag to set unconditional prompt for guidance loss Jaret Burkett 2025-06-24 08:51:29 -06:00
• ba1274d99e Added a guidance burning loss. Modified DFE to work with new model. Bug fixes Jaret Burkett 2025-06-23 08:38:27 -06:00
• 8602470952 Updated diffusion feature extractor Jaret Burkett 2025-06-19 15:36:10 -06:00
• 4586eb5392 Added social links to sidebar Jaret Burkett 2025-06-17 13:25:24 -06:00
• 989ebfaa11 Added a basic torch profiler that can be used in config during development to find some obvious issues. Jaret Burkett 2025-06-17 13:03:39 -06:00
• ff617fdaea Started doing info bubble docs on the simple ui Jaret Burkett 2025-06-17 11:00:24 -06:00
• 595a6f1735 Initial setup for a cron working on the ui for various tasks Jaret Burkett 2025-06-17 07:43:34 -06:00
• 1cc663a664 Performance optimizations for pre processing the batch Jaret Burkett 2025-06-17 07:37:41 -06:00
• 11f2eee53a Hide control images from ui image viewer Jaret Burkett 2025-06-16 07:18:43 -06:00
• 1c2b7298dd More work on mean flow loss. Moved it to an adapter. Still not functioning properly though. Jaret Burkett 2025-06-16 07:17:35 -06:00
• c0314ba325 Fixed some issues with training mean flow algo. Still testing WIP Jaret Burkett 2025-06-14 12:24:00 -06:00
• cbf04b8d53 Fixed some issues with training mean flow algo. Still testing WIP Jaret Burkett 2025-06-14 12:24:00 -06:00
• 0946a66576 Merge branch 'main' into dev Jaret Burkett 2025-06-12 08:11:19 -06:00
• 3f0ae99d48 Version bump Jaret Burkett 2025-06-12 08:01:26 -06:00
• fc83eb7691 WIP on mean flow loss. Still a WIP. Jaret Burkett 2025-06-12 08:00:51 -06:00
• cf11f128b9 Merge pull request #304 from hameerabbasi/fix-caption-loads Jaret Burkett 2025-06-12 07:44:12 -06:00
• 5e86139e0a Fix NameError. Hameer Abbasi 2025-06-11 15:07:20 +02:00
• c5d6b74fea Fix caption loading. Hameer Abbasi 2025-06-11 14:47:26 +02:00
• ba5196dd4a Merge branch 'main' into dev Jaret Burkett 2025-06-10 10:26:11 -06:00
• ffb5fe0667 Version bump Jaret Burkett 2025-06-10 10:04:32 -06:00
• f8fb3b9c45 Added support for sdxl and sd1.5 to the ui. Jaret Burkett 2025-06-10 10:03:54 -06:00
• d5c547da43 Fixed DOP typo Jaret Burkett 2025-06-10 08:44:47 -06:00
• f19f7f9486 Fixed issue with wan2.1 training in ui. Name had a typo Jaret Burkett 2025-06-10 08:42:18 -06:00
• 7317ed58af Adjust the ui of the sidebar Jaret Burkett 2025-06-10 08:40:51 -06:00
• 517bc294fa Update support link Jaret Burkett 2025-06-10 08:28:30 -06:00
• 97e101522c Increase ema feedback amount. Normalize the dfe 4 image embeds Jaret Burkett 2025-06-10 08:01:13 -06:00
• eefa93f16e Various code to support experiments. Jaret Burkett 2025-06-09 11:19:21 -06:00
• 22cdfadab6 Added new timestep weighing strategy Jaret Burkett 2025-06-04 01:16:02 -06:00
• adc31ec77d Small updates and bug fixes for various things Jaret Burkett 2025-06-03 20:08:35 -06:00
• 82b90b902e Double tap torch install to force blackwell compatability Jaret Burkett 2025-06-02 20:22:22 -06:00
• 85f4b47e79 Fix issue with setup tools requirements Jaret Burkett 2025-06-02 06:54:35 -06:00
• e20a869dc1 Update docker install for blacwell Jaret Burkett 2025-06-01 19:06:53 -06:00
• 12fa109910 Updated cuda arch list on docker build Jaret Burkett 2025-06-01 13:37:11 -06:00
• 7d76165dcf Merge branch 'main' into dev Jaret Burkett 2025-06-01 13:33:40 -06:00
• b6d25fcd10 Improvements to vae trainer. Adjust denoise prediction of DFE v3 Jaret Burkett 2025-05-30 12:06:47 -06:00
• ffaf2f154a Fix issue with the way chroma handled gradient checkpointing. Jaret Burkett 2025-05-28 08:41:47 -06:00
• 34f4c14cd6 Work on vae trainer Jaret Burkett 2025-05-28 07:42:48 -06:00
• 79bb9be92b Fix issue with saving chroma full finetune. Jaret Burkett 2025-05-28 07:42:30 -06:00
• 79499fa795 Allow fine tuning pruned versions of chroma. Allow flash attention 2 for chroma if it is installed. Jaret Burkett 2025-05-21 07:02:50 -06:00
• 48e11cf843 Fallback unwrapping logic if fails Jaret Burkett 2025-05-21 03:10:33 -06:00
• 7045a01375 Fixed issue saving optimizer in some instances. Jaret Burkett 2025-05-21 02:27:55 -06:00
• fca7fd6c38 Merge branch 'main' of github.com:ostris/ai-toolkit Jaret Burkett 2025-05-21 02:20:06 -06:00
• e5181d23cd Added some experimental training techniques. Ignore for now. Still in testing. Jaret Burkett 2025-05-21 02:19:54 -06:00
• 4f896c0d8a Fixed issue where sampling fails if doing a full finetune for some models Jaret Burkett 2025-05-17 19:37:55 +00:00
• 01101be196 version bump Jaret Burkett 2025-05-17 05:50:12 -06:00
• 6174ba474e Fixed issue with chroma sampling Jaret Burkett 2025-05-10 18:30:23 +00:00
• 64130189ce Bumped torch and cuda to support blackwell arch Jaret Burkett 2025-05-09 11:17:58 -06:00
• 66a41e49d9 Bump version Jaret Burkett 2025-05-08 17:37:28 -06:00
• 1210050ead Reworked control generator. It is now significantly faster. Also uses better pose model with better license. Jaret Burkett 2025-05-08 14:35:55 -06:00
• 25e150b370 Added support for Flex.2 in the UI Jaret Burkett 2025-05-07 12:41:51 -06:00
• 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 Jaret Burkett 2025-05-07 12:06:30 -06:00
• d9700bdb99 Added initial support for f-lite model Jaret Burkett 2025-05-01 11:15:18 -06:00
• 5890e67a46 Various bug fixes Jaret Burkett 2025-04-29 09:30:33 -06:00
• 2b4c525489 Reworked automagic optimizer and did more testing. Starting to really like it. Working well. Jaret Burkett 2025-04-28 08:01:10 -06:00
• 88b3fbae37 Various experiments and minor bug fixes for edge cases Jaret Burkett 2025-04-25 13:44:38 -06:00
• 8ff85ba14f Add Flex2 training example Jaret Burkett 2025-04-22 11:59:47 -06:00
• 80f73ce9c0 Update README.md Jaret Burkett 2025-04-22 10:46:51 -06:00
• 9f42944056 Update Sponsors Jaret Burkett 2025-04-22 09:44:56 -06:00
• add83df5cc Fixed issue with training hidream when batch size is larger than 1 Jaret Burkett 2025-04-21 17:26:29 +00:00
• 12e3095d8a Fixed issue with saving base model version Jaret Burkett 2025-04-19 14:34:01 -06:00
• 77001ee77f Upodate model tag on loras Jaret Burkett 2025-04-19 10:41:27 -06:00
• d455e76c4f Cleanup Jaret Burkett 2025-04-18 11:44:49 -06:00
• 1628884254 Remove submodule install from docker Jaret Burkett 2025-04-18 10:41:52 -06:00
• 9c422ac14f Bump version Jaret Burkett 2025-04-18 10:39:51 -06:00
• bfe29e2151 Removed all submodules. Submodule free now, yay. Jaret Burkett 2025-04-18 10:39:15 -06:00
• bd2de5b74e Remove leco submodule Jaret Burkett 2025-04-18 10:08:09 -06:00
• 970fac19a5 Remove batch annotator as submodule Jaret Burkett 2025-04-18 10:03:37 -06:00
• 5f312cd46b Remove ip adapter submodule Jaret Burkett 2025-04-18 09:59:42 -06:00
• c90615f8bb Add model hooks to polarity loss Jaret Burkett 2025-04-17 09:00:10 -06:00
• 5961ef6c9f Fixed typo in linux install Jaret Burkett 2025-04-16 22:08:17 -06:00
• fd6026ab73 Merge pull request #278 from ostris/hidream Jaret Burkett 2025-04-16 13:48:27 -06:00
• 79c87701e7 Add hidream to the ui Jaret Burkett 2025-04-16 13:45:21 -06:00
• fecc64e646 Update hidream defaults, pass additional information to flow guidance Jaret Burkett 2025-04-16 13:03:04 -06:00
• d5a64006b5 Added example config to train hidream Jaret Burkett 2025-04-16 10:18:22 -06:00
• 0f99fce004 Adjust hidream lora names to work with comfy Jaret Burkett 2025-04-16 09:24:23 -06:00
• c12036df95 Added ability to use short captions from json caption file Jaret Burkett 2025-04-16 08:32:28 -06:00
• 68018c908e Made a script to convert diffusers to comfy just the transformer Jaret Burkett 2025-04-15 10:22:05 -06:00
• 524bd2edfc Make flash attn optional. Handle larger batch sizes. Jaret Burkett 2025-04-14 14:34:46 +00:00
• 89c0f688db Merge branch 'main' into hidream Jaret Burkett 2025-04-13 21:16:07 -06:00
• 1e0bff653c Fix new bug I accidently introduced with lora Jaret Burkett 2025-04-13 21:15:07 -06:00
• 3a5ea2c742 Remove some moe stuff for finetuning. Drastically reduces vram usage Jaret Burkett 2025-04-14 00:57:34 +00:00
• f80cf99f40 Hidream is training, but has a memory leak Jaret Burkett 2025-04-13 23:28:18 +00:00
• 594e166ca3 Initial support for hidream. Still a WIP Jaret Burkett 2025-04-13 13:50:11 -06:00
• ca3ce0f34c Make it easier to designate lora blocks for new models. Improve i2v adapter speed. Fix issue with i2v adapter where cached torch tensor was wrong range. Jaret Burkett 2025-04-13 13:49:13 -06:00
• 6fb44db6a0 Finished up first frame for i2v adapter Jaret Burkett 2025-04-12 17:13:04 -06:00
• cd37ccfc2e Use gradient checkpointing on DFE models if set Jaret Burkett 2025-04-11 10:45:39 -06:00
• 4a43589666 Use a shuffled embedding as unconditional for i2v adapter Jaret Burkett 2025-04-11 10:44:43 -06:00
• 059155174a Added mask diffirential mask dialation for flex2. Handle video for the i2v adapter Jaret Burkett 2025-04-10 11:50:01 -06:00
• 9794416a5d Fixed bug when loading video datasets Jaret Burkett 2025-04-10 08:16:05 -06:00
• d8bdc03256 Allow full control of caption extensions Jaret Burkett 2025-04-10 07:42:04 -06:00
• 96ba2fd129 Added methods to the dataloader to automatically generate controls for line, mask, inpainting, depth, and pose. Jaret Burkett 2025-04-09 13:35:04 -06:00