Jaret Burkett
|
f38de2a2fe
|
Add tipsv2 locally and fix gradient checkpointing for it
|
2026-05-10 14:47:44 -06:00 |
|
Jaret Burkett
|
d144cb5ea6
|
Switch to uplot for loss graph and rework performance of graph. It is significantly more perfromant now.
|
2026-05-07 07:38:00 -06:00 |
|
Jaret Burkett
|
a12ddd72a1
|
Change the velocity weight cap on dfe 9
|
2026-05-07 07:37:05 -06:00 |
|
Jaret Burkett
|
6bb8acbffc
|
Add agreement_threshold default of 0.6 to automagic 2
|
2026-05-05 19:13:00 -06:00 |
|
Jaret Burkett
|
963a9f42b2
|
Add decode latent to wan 2.1 models. Add gradinet checkpointing to wan vae.
|
2026-05-05 11:30:16 -06:00 |
|
Jaret Burkett
|
4260a3c5b6
|
Add optimizer test suite and make minor speed adjustments to Automagicv2
|
2026-05-05 10:02:30 -06:00 |
|
Jaret Burkett
|
aeca7fe404
|
Add Automagic v2 optimizer. It uses significantly less vram and is much more efficient.
|
2026-05-05 09:09:07 -06:00 |
|
Jaret Burkett
|
0d91fcee9e
|
Allow edit of captioning job
|
2026-04-30 16:00:16 -06:00 |
|
Jaret Burkett
|
eadc9a58af
|
Version bump
|
2026-04-30 06:00:58 -06:00 |
|
Jaret Burkett
|
e9ab387dfd
|
Fixed issue with qwen image edit models when using multiple control images when not caching text embeddings.
|
2026-04-30 11:59:12 +00:00 |
|
Jaret Burkett
|
deb409085a
|
Made it possible to use the flux 2 small decoder VAE when setting the vae_path manually for flux2 models
|
2026-04-30 05:42:31 -06:00 |
|
Jaret Burkett
|
7ccec8ec2c
|
Add checkpointing and a proper decode for flux 2 VAEs so they can be used with DFE
|
2026-04-30 04:27:13 -06:00 |
|
Jaret Burkett
|
b4f0efb025
|
Version Bump
|
2026-04-28 13:42:00 -06:00 |
|
Jaret Burkett
|
af6458d1b5
|
Enable caching of ACE step latents.
|
2026-04-28 13:39:20 -06:00 |
|
Jaret Burkett
|
77b8765939
|
Fix issue with hover styling on select components.
|
2026-04-28 10:55:20 -06:00 |
|
Jaret Burkett
|
43989cc19e
|
Add advanced config section to the captioner
|
2026-04-28 10:49:46 -06:00 |
|
Jaret Burkett
|
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
|
2026-04-28 10:15:11 -06:00 |
|
Jaret Burkett
|
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.
|
2026-04-28 09:10:02 -06:00 |
|
Jaret Burkett
|
1fc4ad3979
|
Add sapiens2 as a diffusion feature extractor
|
2026-04-27 15:59:03 -06:00 |
|
Jaret Burkett
|
67d67f8c1d
|
Ignore hidden files when captioning
|
2026-04-25 17:38:31 -06:00 |
|
Jaret Burkett
|
998a02f30e
|
Yet another pseudo_huber fix. Still drinking coffee.
|
2026-04-19 10:05:40 -06:00 |
|
Jaret Burkett
|
fc85410c9a
|
Fix issue with precision on pseudo_huber loss
|
2026-04-19 10:02:27 -06:00 |
|
Jaret Burkett
|
20a99258b8
|
Fix issue is is else on pseudo_huber loss
|
2026-04-19 09:59:18 -06:00 |
|
Jaret Burkett
|
f4445cd78c
|
Added psuedo_huber loss
|
2026-04-19 09:51:46 -06:00 |
|
Jaret Burkett
|
488878f354
|
Use hidden layers in the loss for DFE 7 and 8
|
2026-04-18 13:07:38 -06:00 |
|
Jaret Burkett
|
beb40ae29b
|
Add DFE8 with partial step
|
2026-04-17 17:40:16 -06:00 |
|
Jaret Burkett
|
7c4f18ce51
|
Fix ernie unpatchify
|
2026-04-17 12:03:39 -06:00 |
|
Jaret Burkett
|
8cb9649382
|
Add decode latent to ernie pipe
|
2026-04-17 12:01:09 -06:00 |
|
Jaret Burkett
|
67048df9f9
|
Merge branch 'main' of github.com:ostris/ai-toolkit
|
2026-04-17 06:05:40 -06:00 |
|
Jaret Burkett
|
be54094704
|
Version bump
|
2026-04-17 06:05:26 -06:00 |
|
Jaret Burkett
|
a513a1583e
|
Fixed issue where Qwen VL MOE captioner produced nonsense
|
2026-04-16 21:24:43 +00:00 |
|
Jaret Burkett
|
22ea3dd620
|
Fixed issue on some systems where Logger didnt have atty
|
2026-04-16 21:09:52 +00:00 |
|
Jaret Burkett
|
ab1ee4df34
|
Hotfix some issues with Wan models caused by diffusers and transformers updates
|
2026-04-16 20:53:50 +00:00 |
|
Jaret Burkett
|
0c18b39346
|
Version bump
|
2026-04-16 13:09:45 -06:00 |
|
Jaret Burkett
|
afb62b1fa5
|
Add support for Nucleus-Image
|
2026-04-16 13:09:10 -06:00 |
|
Jaret Burkett
|
2faba22b46
|
Fix issue when saving advanced prompt embeds. No such file or directory error
|
2026-04-16 12:22:56 -06:00 |
|
Jaret Burkett
|
0792352dab
|
version bump
|
2026-04-16 09:25:14 -06:00 |
|
Jaret Burkett
|
8f67f5022e
|
Version lock peft version
|
2026-04-15 14:17:26 -06:00 |
|
Jaret Burkett
|
acc3e60140
|
Add ernie to readme
|
2026-04-15 06:43:36 -06:00 |
|
Jaret Burkett
|
dd7074a21f
|
Fix issue with layer offloading on ernie
|
2026-04-14 19:19:02 -06:00 |
|
Jaret Burkett
|
e74bc9ac7b
|
Fix issue with concatinating advanced prompt embeds.
|
2026-04-14 16:04:34 -06:00 |
|
Jaret Burkett
|
7eb1226a6d
|
Fix issue with loading advanced prompt configs metadata
|
2026-04-14 15:42:09 -06:00 |
|
Jaret Burkett
|
97d8c05d75
|
Fix issue with captioner sometimes outputting not utf-8 characters
|
2026-04-14 14:37:21 -06:00 |
|
Jaret Burkett (Ostris)
|
3e0c904054
|
Add support for Baidu's ERNIE-Image (#793)
* Add support for ERNIE Image
* change float64 to float32
* Version bump
* Update ERNIE defaults
|
2026-04-14 09:45:12 -06:00 |
|
Asaf Agami
|
e868fca562
|
fix custom_flowmatch_sampler (#783)
|
2026-04-13 09:42:05 -06:00 |
|
Jaret Burkett
|
233e292256
|
Added some experimental low step things for zeta
|
2026-04-13 09:37:34 -06:00 |
|
Jaret Burkett
|
1058ef3513
|
Made AdvancedPromptEmbeds that is compatable with previous PromptEmbeds functionality, but is more streamlined and can accomidate more model embedding paradigms.
|
2026-04-11 10:45:47 -06:00 |
|
Jaret Burkett
|
0d11be41fa
|
Adjust default sample steps.
|
2026-04-10 17:46:34 -06:00 |
|
Jaret Burkett
|
62e18427b4
|
Adjust default sample steps.
|
2026-04-10 17:46:09 -06:00 |
|
Jaret Burkett
|
9b4e2d1b0b
|
More flac support
|
2026-04-10 12:27:09 -06:00 |
|