Jaret Burkett
7a089fd0d7
Add support for training directly on Krea2 Turbo with a training adapter
2026-06-23 20:32:02 -06:00
Jaret Burkett (Ostris)
99be3d96a2
Add support for Krea2 ( #906 )
...
* Add support for krea2
* Update repo pointer to actual repo
2026-06-23 09:18:14 -06:00
Jaret Burkett
dba092fc15
Keep a real git repo in the docker image
2026-06-21 06:54:19 -06:00
Jaret Burkett
60c1ac6a50
Add support for Boogu Image and Boogu Image Edit
2026-06-18 15:05:49 -06:00
Jaret Burkett
92814f9e6d
Apply ideogram dynamic shifting to sampling
2026-06-16 18:52:16 -06:00
Jaret Burkett
178eb5fbbe
Add unconditional lora support so Ideogram 4 inference will more closely resemble the full pipeline results. I pushed a finetuned unconditional lora to the hub as an adapter.
2026-06-16 13:27:43 -06:00
Jaret Burkett
86b19589a0
Update the Ideogram 4 prompt generation/parsing/ui to handle the updated format notes better.
2026-06-16 09:44:38 -06:00
Jaret Burkett
5df3fb69e3
Rework Docker image for a minimal build/pull/push footprint
2026-06-14 12:19:58 -06:00
Jaret Burkett
17c9279828
Version bump
2026-06-13 09:48:26 -06:00
Jaret Burkett
d83f7dd4d9
Fix a few issues with compile. Changed defaults. Future proofed block layer compile.
2026-06-12 11:43:43 -06:00
Jaret Burkett
a5558ae7d9
Add compile to the actual right config section.
2026-06-12 11:12:41 -06:00
Jaret Burkett
c09b228a35
Add model compiling to the ui
2026-06-12 11:02:10 -06:00
Jaret Burkett
c9cdbb5bb7
Version bump
2026-06-07 16:08:11 -06:00
Jaret Burkett
fa0dca288d
Version bump
2026-06-06 10:24:31 -06:00
Jaret Burkett
10cdeb394e
Version bump
2026-06-05 14:09:46 -06:00
Jaret Burkett
6fd2253932
Imporved editing boxed on the dataset viewer
2026-06-05 10:30:06 -06:00
Jaret Burkett
edcc7415d1
Added an autocaptioner for Ideogram 4 captions
2026-06-04 13:04:23 -06:00
Jaret Burkett
2ddc2e1318
Update the default Ideogram 4 prompts to work significantly better.
2026-06-04 10:32:05 -06:00
Jaret Burkett
63b3181262
Added experimental support for Ideogram 4
2026-06-04 09:03:00 -06:00
Jaret Burkett
f5446c0d5f
Version bump
2026-06-01 02:16:47 -06:00
Jaret Burkett
68e9b38220
Add ability to trigger a save from the ui which whill make the trainer save on the next step
2026-05-26 08:38:49 -06:00
Jaret Burkett
6102370df9
Add support for ZImage L2P
2026-05-22 14:58:28 -06:00
Jaret Burkett
df851b3497
Made the UI mobile friendly, finally...
2026-05-19 09:27:36 -06:00
Jaret Burkett
6ecaf679dc
Add ability to run small scripts from the ui and added a merge lora script
2026-05-18 14:33:47 -06:00
Jaret Burkett (Ostris)
fc4d6ebf39
Add support for fine-tuning Hidream O1 ( #831 )
...
* Initial support for hidream. Lora keys likely need work
* Fix saving for hidream-o1
* Remove dependence on flash attention for hidream o1
* Fix gradient checkpointing for hidream o1
* A lot of fixes for hidream. Handle loading and saving as comfy model.
* Omit layers not used in comfy. Fix issue with lora loading keys in comfy
* Version bumpo
2026-05-12 11:15:16 -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
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
eadc9a58af
Version bump
2026-04-30 06:00:58 -06:00
Jaret Burkett
b4f0efb025
Version Bump
2026-04-28 13:42:00 -06:00
Jaret Burkett
be54094704
Version bump
2026-04-17 06:05:26 -06:00
Jaret Burkett
0c18b39346
Version bump
2026-04-16 13:09:45 -06:00
Jaret Burkett
0792352dab
version bump
2026-04-16 09:25:14 -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
Jaret Burkett
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
Jaret Burkett
853ffaf207
Add light mode support.
2026-03-31 16:54:55 -06:00
Jaret Burkett
dcb7f465ec
Version Bump
2026-03-30 15:57:40 -06:00
Jaret Burkett
8302b21f8f
Version Bump
2026-03-28 13:23:52 -06:00
Jaret Burkett
5ce87fa48b
Version bump
2026-03-27 20:26:31 -06:00
Jaret Burkett
f85bf065bf
Use pooler embeddings for DFE v6 with dino v3
2026-03-27 07:02:07 -06:00
Jaret Burkett
b8000dbcbc
Bump version
2026-03-25 08:18:42 -06:00
Jaret Burkett
7f3309b291
Add support for audo frame count so datasets can have varrying length videos. Varous ltx 2.3 VAE optimizations such as removing tiling articacts, and doing frame split encoding to reduce vram on encoding/decoding.
2026-03-24 12:20:09 -06:00
Jaret Burkett
5642b656b9
Fix audio issues with ltx2 models. Silent codec fails now raised. Auto convert surround sound audio to stereo. Invalidate old caches just to be safe so they recache now.
2026-03-23 20:08:33 +00:00
Jaret Burkett
330059d8a1
version bump
2026-03-23 11:01:16 -06:00
Jaret Burkett
253cb31362
Fix issue with video and images with no audio on ltx models
2026-03-22 22:09:23 -06:00
Jaret Burkett
4a3d317e2b
Fix issue with using the default text encoder with ltx 2.3
2026-03-22 18:53:59 -06:00
Jaret Burkett
859635e95b
Add support for training LTX 2.3 ( #745 )
...
* Initial support for ltx 2.3. Still needs a lot of testing to make sure it is all right.
* bump version
* Handle lora renaming keys for new ltx 2.3 layers
2026-03-22 17:56:59 -06:00
Jaret Burkett
57d407cfd4
Add support for training lodestones/Zeta-Chroma
2026-03-01 12:52:29 -07:00
Jaret Burkett
de7d22c9be
Version bump
2026-02-19 11:58:15 -07:00
Jaret Burkett
50664c2421
Version bump
2026-01-28 12:55:32 -07:00
Jaret Burkett
2db090144a
Add support for Z-Image
2026-01-27 09:34:46 -07:00