fc85410c9a
Fix issue with precision on pseudo_huber loss
2026-04-19 10:02:27 -06:00
20a99258b8
Fix issue is is else on pseudo_huber loss
2026-04-19 09:59:18 -06:00
f4445cd78c
Added psuedo_huber loss
2026-04-19 09:51:46 -06:00
488878f354
Use hidden layers in the loss for DFE 7 and 8
2026-04-18 13:07:38 -06:00
beb40ae29b
Add DFE8 with partial step
2026-04-17 17:40:16 -06:00
7c4f18ce51
Fix ernie unpatchify
2026-04-17 12:03:39 -06:00
8cb9649382
Add decode latent to ernie pipe
2026-04-17 12:01:09 -06:00
67048df9f9
Merge branch 'main' of github.com:ostris/ai-toolkit
2026-04-17 06:05:40 -06:00
be54094704
Version bump
2026-04-17 06:05:26 -06:00
a513a1583e
Fixed issue where Qwen VL MOE captioner produced nonsense
2026-04-16 21:24:43 +00:00
22ea3dd620
Fixed issue on some systems where Logger didnt have atty
2026-04-16 21:09:52 +00:00
ab1ee4df34
Hotfix some issues with Wan models caused by diffusers and transformers updates
2026-04-16 20:53:50 +00:00
0c18b39346
Version bump
2026-04-16 13:09:45 -06:00
afb62b1fa5
Add support for Nucleus-Image
2026-04-16 13:09:10 -06:00
2faba22b46
Fix issue when saving advanced prompt embeds. No such file or directory error
2026-04-16 12:22:56 -06:00
0792352dab
version bump
2026-04-16 09:25:14 -06:00
8f67f5022e
Version lock peft version
2026-04-15 14:17:26 -06:00
acc3e60140
Add ernie to readme
2026-04-15 06:43:36 -06:00
dd7074a21f
Fix issue with layer offloading on ernie
2026-04-14 19:19:02 -06:00
e74bc9ac7b
Fix issue with concatinating advanced prompt embeds.
2026-04-14 16:04:34 -06:00
7eb1226a6d
Fix issue with loading advanced prompt configs metadata
2026-04-14 15:42:09 -06:00
97d8c05d75
Fix issue with captioner sometimes outputting not utf-8 characters
2026-04-14 14:37:21 -06:00
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
e868fca562
fix custom_flowmatch_sampler ( #783 )
2026-04-13 09:42:05 -06:00
233e292256
Added some experimental low step things for zeta
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.
2026-04-11 10:45:47 -06:00
0d11be41fa
Adjust default sample steps.
2026-04-10 17:46:34 -06:00
62e18427b4
Adjust default sample steps.
2026-04-10 17:46:09 -06:00
9b4e2d1b0b
More flac support
2026-04-10 12:27:09 -06:00
0b9c365acb
Add flac and ogg support
2026-04-10 12:10:54 -06:00
bfb373c8fa
Prep for future breaking changes in newer versions of transformers library
2026-04-10 12:04:32 -06:00
145144eee3
Fix issue with auto updating captions when captioning a dataset
2026-04-10 11:51:18 -06:00
765a9d5b2e
Add a download button to music samples in in the gear menu
2026-04-10 10:07:45 -06:00
d08ea8318f
Change default timestep type for ace step to linear
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 )
...
* Base ace step 1.5 xl added. Generating, still wip on training and ui
* Base training code done
* Fix some issues with caching text embeddings. Update sample cards to show audio
* Fix issue with quantizing ace step
* Add album artwork to samples with waveform.
* Cleanup logs
* Add album art endpoint to speed up album art loading
* Made an make video with artwork script
* Make ui handle basic audio models. Make multi line adjustments to the editor and better syntax hilighting.
* Add prompt tagging system for special tagged models.
* prompt tagging processing for ui working.
* Moved default samples to a special file so we can add more when needed and they can be adjusted for a specific model
* Add a captioner job with music captioner that is prepped for use with the ui
* Add basit ui setup for captioning modal and handeling captioning jobs
* Starting captioning job from ui working. Still better management for it.
* Better filtering of job options in the job view for captioning jobs
* Added qwen3 vl as a captioner for images
* Have an indicator when a dataset is being captioned.
* Adjust the way caption jobs look in the queue
* Fix a few issues. Adjust defaults.
* Version bump
* Added ace step to the readme.
2026-04-09 15:02:03 -06:00
9ca58e9aa2
Fixed offload and quantize order of ltx 2.3 text encoder.
2026-04-07 15:11:50 -06:00
0dcbabf6af
Fix merge nertwork ref
2026-04-01 10:38:31 -06:00
f213e3b1e5
Fix FLUX2 Klein load-time VRAM spikes on low-memory GPUs. ( #726 )
...
Keep the transformer and Qwen text encoder off CUDA during initial load/quantization in low-VRAM mode so model startup avoids full-model OOM before offloading and quantization can take effect.
Co-authored-by: Cursor <cursoragent@cursor.com>
Co-authored-by: Jaret Burkett <jaretburkett@gmail.com>
2026-04-01 09:36:55 -06:00
da2a79590f
Add a merge network on save strength
2026-04-01 09:21:08 -06:00
853ffaf207
Add light mode support.
2026-03-31 16:54:55 -06:00
ad474e3d06
Update and reformat the readme
2026-03-31 12:31:05 -06:00
4a3251640a
More work on compiling models
2026-03-31 12:11:56 -06:00
358d684f6f
Move compiiling the model after accelerate manipulation
2026-03-31 09:52:27 -06:00
0045260af7
Fix issue where compile true did not actually compile the model
2026-03-31 09:27:54 -06:00
e22039e4aa
Add more optimizers to the ui
2026-03-31 09:20:20 -06:00
bf56217c37
Fixed issue where job would fail if DB is locked.
2026-03-31 09:10:33 -06:00
dcb7f465ec
Version Bump
2026-03-30 15:57:40 -06:00
626d9674ea
Add info about automated coding agent pull requests. So sick of them.
2026-03-30 11:02:12 -06:00
b43ea6c2d3
Abort caption requests when they are not in view to tax the server less.
2026-03-30 10:46:25 -06:00
a484e55d66
Rework dataset model and file dragging. Use single model for dragging, uploading, and selecting images.
2026-03-30 10:33:10 -06:00
Jaret Burkett
Asaf Agami
M. Hofer