Commit Graph

Select branches

Hide Pull Requests

evan_dev

main

Mono Color

• b5f21ae695 Fixed issue where the training type dropdown in the ui was not showing fully Jaret Burkett 2026-06-03 10:14:15 -06:00
• d9f26c2f87 Add gradient checkpointing to tipsv2 heads Jaret Burkett 2026-06-01 05:00:15 -06:00
• bd468727a6 Fixed issue where some image cards in samples and datasets would show gray instead of the image until you scroll. Jaret Burkett 2026-06-01 03:36:32 -06:00
• f5446c0d5f Version bump Jaret Burkett 2026-06-01 02:16:47 -06:00
• e5439509b5 Added pure lpips dfe Jaret Burkett 2026-05-31 11:52:03 -06:00
• 212cfe998a Add ability to download or delete the optimizer state from the ui Jaret Burkett 2026-05-29 08:56:22 -06:00
• 5e84bf0d0b Fixed issue with hidream-01 that could cause a weird nan state. Took forever to track down as it was 1 in 10 starts. Jaret Burkett 2026-05-28 12:48:24 -06:00
• 87bac27513 Fixed issue with new bucket scaler Jaret Burkett 2026-05-28 11:34:16 -06:00
• 30886b8f92 Added a wallet indicator for Ostris Cloud Jaret Burkett 2026-05-28 11:09:51 -06:00
• 3e86d81fc6 Adjust bucket sizes to achieve maximum pixels without going over. Jaret Burkett 2026-05-28 09:36:17 -06:00
• ef57c1077c Change z image divisibility Jaret Burkett 2026-05-28 09:09:36 -06:00
• 15082cfb8a Round buckets for divisibility instead of always rounding down. Jaret Burkett 2026-05-28 09:08:21 -06:00
• c9264bdd0b Reworked the bucketing system to precisly match model specific divisibility. The old SDXL bucket sizes needed to go. Jaret Burkett 2026-05-28 08:35:58 -06:00
• 68e9b38220 Add ability to trigger a save from the ui which whill make the trainer save on the next step Jaret Burkett 2026-05-26 08:38:49 -06:00
• 2aa60e4ca5 Update default agreement threshold for automagic v2 to be 0.5 Jaret Burkett 2026-05-26 07:57:54 -06:00
• 76c99da4e4 Add version to the ui Jaret Burkett 2026-05-26 07:30:09 -06:00
• 266956068a Add a way to delete checkpoints on the ui Jaret Burkett 2026-05-26 07:09:21 -06:00
• 954c5efec8 Show job info on sidebar active job in ui Jaret Burkett 2026-05-25 11:07:29 -06:00
• 083236a2a7 Drastically improve the loading speed of images in the ui by using a custom loader and abort controller to abort when images leave the view. Jaret Burkett 2026-05-25 10:04:25 -06:00
• 7354def271 Added decode latent method to qwen image model Jaret Burkett 2026-05-25 10:02:09 -06:00
• 3d836ac371 Show active jobs in the sidebar of the ui Jaret Burkett 2026-05-25 08:21:42 -06:00
• a798e06dd2 Updated the support button to look more like a button Jaret Burkett 2026-05-25 07:48:04 -06:00
• 8042cbe9d2 Added virtulization for sample images to handle huge number of samples more efficientyly. Jaret Burkett 2026-05-25 07:22:24 -06:00
• fbac1cb7f5 Dont force flash attention on hidream 01. Causes random issues and is slower. Jaret Burkett 2026-05-24 16:05:22 -06:00
• 307ff11bc5 Drastically improved the performance of the dataset viewer on large datasets by switching to virtualization. Fixed issue with images loading when they have two periods in a row .. Jaret Burkett 2026-05-24 15:13:55 -06:00
• c6a7e81a70 Added a dataset image viewer. Fixed an issue where captions would show as not saved when they were saved. Jaret Burkett 2026-05-24 14:31:06 -06:00
• 12304e170f Added some experimental loss targets Jaret Burkett 2026-05-24 14:13:23 -06:00
• 644a6f9246 Fix device casting for zimage in some instances Jaret Burkett 2026-05-23 10:37:33 -06:00
• c6ecc03ccd Fixed saving full model of z_image l2p for finetuning Jaret Burkett 2026-05-23 07:36:24 -06:00
• 6102370df9 Add support for ZImage L2P Jaret Burkett 2026-05-22 14:58:28 -06:00
• 6fc08a8928 Fix issue with saving images from new sample viewer Jaret Burkett 2026-05-22 11:58:32 -06:00
• 5579837c3f Add hidream o1 to the readme Jaret Burkett 2026-05-21 07:56:54 -06:00
• aecd554128 Add sapiens2 matting as a mask generator. Begin transition to model paths and model folders. Jaret Burkett 2026-05-20 08:56:16 -06:00
• 15d4fb89ff Fixed overflow issue on modal Jaret Burkett 2026-05-19 10:12:43 -06:00
• df851b3497 Made the UI mobile friendly, finally... Jaret Burkett 2026-05-19 09:27:36 -06:00
• 6ecaf679dc Add ability to run small scripts from the ui and added a merge lora script Jaret Burkett 2026-05-18 14:33:47 -06:00
• ec58dcde92 Add better gradient checkpointing to dfes Jaret Burkett 2026-05-18 09:25:23 -06:00
• b42acb988f Remove future steps from loss log if resuming from an earlier step Jaret Burkett 2026-05-18 09:23:55 -06:00
• e03c6e4dc9 Fix potential inconsistency with different attention mentods in hidream01 Jaret Burkett 2026-05-13 09:08:11 -06:00
• 4bfe944792 Scale dfe 7 with velocity_equiv_weight Jaret Burkett 2026-05-13 09:06:59 -06:00
• fc4d6ebf39 Add support for fine-tuning Hidream O1 (#831) Jaret Burkett (Ostris) 2026-05-12 11:15:16 -06:00
• f38de2a2fe Add tipsv2 locally and fix gradient checkpointing for it Jaret Burkett 2026-05-10 14:47:44 -06:00
• d144cb5ea6 Switch to uplot for loss graph and rework performance of graph. It is significantly more perfromant now. Jaret Burkett 2026-05-07 07:38:00 -06:00
• a12ddd72a1 Change the velocity weight cap on dfe 9 Jaret Burkett 2026-05-07 07:37:05 -06:00
• 6bb8acbffc Add agreement_threshold default of 0.6 to automagic 2 Jaret Burkett 2026-05-05 19:13:00 -06:00
• 963a9f42b2 Add decode latent to wan 2.1 models. Add gradinet checkpointing to wan vae. Jaret Burkett 2026-05-05 11:30:16 -06:00
• 4260a3c5b6 Add optimizer test suite and make minor speed adjustments to Automagicv2 Jaret Burkett 2026-05-05 10:02:30 -06:00
• aeca7fe404 Add Automagic v2 optimizer. It uses significantly less vram and is much more efficient. Jaret Burkett 2026-05-05 09:09:07 -06:00
• 0c8793ff47 fix: make pin_memory opt-in via dataset config (default off) Booyaka101 2026-05-02 11:02:41 +08:00
• 67e223b502 perf: enable pin_memory on CUDA DataLoaders for faster CPU->GPU transfer Booyaka101 2026-05-01 23:07:43 +08:00
• 0d91fcee9e Allow edit of captioning job Jaret Burkett 2026-04-30 16:00:16 -06:00
• eadc9a58af Version bump Jaret Burkett 2026-04-30 06:00:58 -06:00
• e9ab387dfd Fixed issue with qwen image edit models when using multiple control images when not caching text embeddings. Jaret Burkett 2026-04-30 11:59:12 +00:00
• deb409085a Made it possible to use the flux 2 small decoder VAE when setting the vae_path manually for flux2 models Jaret Burkett 2026-04-30 05:42:31 -06:00
• 7ccec8ec2c Add checkpointing and a proper decode for flux 2 VAEs so they can be used with DFE Jaret Burkett 2026-04-30 04:27:13 -06:00
• b4f0efb025 Version Bump Jaret Burkett 2026-04-28 13:42:00 -06:00
• af6458d1b5 Enable caching of ACE step latents. Jaret Burkett 2026-04-28 13:39:20 -06:00
• 77b8765939 Fix issue with hover styling on select components. Jaret Burkett 2026-04-28 10:55:20 -06:00
• 43989cc19e Add advanced config section to the captioner Jaret Burkett 2026-04-28 10:49:46 -06:00
• f972b750e6 Performance improvements to captioner. Add ability to set a default caption for ace captioner to avoid needing to caption the audio so we only transcribe Jaret Burkett 2026-04-28 10:15:11 -06:00
• acc6a36214 Scale DFE 9 to a velocity equiv weight to match flow matching gradient strength. Probably need to rework all DFEs to do this as the math checks out. Jaret Burkett 2026-04-28 09:10:02 -06:00
• 1fc4ad3979 Add sapiens2 as a diffusion feature extractor Jaret Burkett 2026-04-27 15:59:03 -06:00
• 67d67f8c1d Ignore hidden files when captioning Jaret Burkett 2026-04-25 17:38:31 -06:00
• 998a02f30e Yet another pseudo_huber fix. Still drinking coffee. Jaret Burkett 2026-04-19 10:05:40 -06:00
• fc85410c9a Fix issue with precision on pseudo_huber loss Jaret Burkett 2026-04-19 10:02:27 -06:00
• 20a99258b8 Fix issue is is else on pseudo_huber loss Jaret Burkett 2026-04-19 09:59:18 -06:00
• f4445cd78c Added psuedo_huber loss Jaret Burkett 2026-04-19 09:51:46 -06:00
• 488878f354 Use hidden layers in the loss for DFE 7 and 8 Jaret Burkett 2026-04-18 13:07:38 -06:00
• beb40ae29b Add DFE8 with partial step Jaret Burkett 2026-04-17 17:40:16 -06:00
• 7c4f18ce51 Fix ernie unpatchify Jaret Burkett 2026-04-17 12:03:39 -06:00
• 8cb9649382 Add decode latent to ernie pipe Jaret Burkett 2026-04-17 12:01:09 -06:00
• 67048df9f9 Merge branch 'main' of github.com:ostris/ai-toolkit Jaret Burkett 2026-04-17 06:05:40 -06:00
• be54094704 Version bump Jaret Burkett 2026-04-17 06:05:26 -06:00
• a513a1583e Fixed issue where Qwen VL MOE captioner produced nonsense Jaret Burkett 2026-04-16 21:24:43 +00:00
• 22ea3dd620 Fixed issue on some systems where Logger didnt have atty Jaret Burkett 2026-04-16 21:09:52 +00:00
• ab1ee4df34 Hotfix some issues with Wan models caused by diffusers and transformers updates Jaret Burkett 2026-04-16 20:53:50 +00:00
• 0c18b39346 Version bump Jaret Burkett 2026-04-16 13:09:45 -06:00
• afb62b1fa5 Add support for Nucleus-Image Jaret Burkett 2026-04-16 13:09:10 -06:00
• 2faba22b46 Fix issue when saving advanced prompt embeds. No such file or directory error Jaret Burkett 2026-04-16 12:22:56 -06:00
• 0792352dab version bump Jaret Burkett 2026-04-16 09:25:14 -06:00
• 8f67f5022e Version lock peft version Jaret Burkett 2026-04-15 14:17:26 -06:00
• acc3e60140 Add ernie to readme Jaret Burkett 2026-04-15 06:43:36 -06:00
• dd7074a21f Fix issue with layer offloading on ernie Jaret Burkett 2026-04-14 19:19:02 -06:00
• e74bc9ac7b Fix issue with concatinating advanced prompt embeds. Jaret Burkett 2026-04-14 16:04:34 -06:00
• 7eb1226a6d Fix issue with loading advanced prompt configs metadata Jaret Burkett 2026-04-14 15:42:09 -06:00
• 97d8c05d75 Fix issue with captioner sometimes outputting not utf-8 characters Jaret Burkett 2026-04-14 14:37:21 -06:00
• 3e0c904054 Add support for Baidu's ERNIE-Image (#793) Jaret Burkett (Ostris) 2026-04-14 09:45:12 -06:00
• e868fca562 fix custom_flowmatch_sampler (#783) Asaf Agami 2026-04-13 18:42:05 +03:00
• 233e292256 Added some experimental low step things for zeta Jaret Burkett 2026-04-13 09:37:34 -06:00
• 1058ef3513 Made AdvancedPromptEmbeds that is compatable with previous PromptEmbeds functionality, but is more streamlined and can accomidate more model embedding paradigms. Jaret Burkett 2026-04-11 10:45:47 -06:00
• 0d11be41fa Adjust default sample steps. Jaret Burkett 2026-04-10 17:46:34 -06:00
• 62e18427b4 Adjust default sample steps. Jaret Burkett 2026-04-10 17:46:09 -06:00
• 9b4e2d1b0b More flac support Jaret Burkett 2026-04-10 12:27:09 -06:00
• 0b9c365acb Add flac and ogg support Jaret Burkett 2026-04-10 12:10:54 -06:00
• bfb373c8fa Prep for future breaking changes in newer versions of transformers library Jaret Burkett 2026-04-10 12:04:32 -06:00
• 145144eee3 Fix issue with auto updating captions when captioning a dataset Jaret Burkett 2026-04-10 11:51:18 -06:00
• 765a9d5b2e Add a download button to music samples in in the gear menu Jaret Burkett 2026-04-10 10:07:45 -06:00
• d08ea8318f Change default timestep type for ace step to linear Jaret Burkett 2026-04-09 17:39:00 -06:00
• 78cf049c29 Add support for ACE-Step 1.5 and ACE-Step 1.5 XL. Also added dataset captioning through the UI. (#785) Jaret Burkett 2026-04-09 15:02:03 -06:00
• 9ca58e9aa2 Fixed offload and quantize order of ltx 2.3 text encoder. Jaret Burkett 2026-04-07 15:11:50 -06:00