Commit Graph

115 Commits

Author SHA1 Message Date
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
Jaret Burkett a6da9e37ac Add support for FLUX.2 klein base models 2026-01-17 17:46:25 -07:00
Jaret Burkett 73dedbf662 Do caching of latents, first frame and audio when caching latents for LTX2 2026-01-14 11:05:23 -07:00
Jaret Burkett 64fe29b182 Support img 2 vid training for ltx-2 2026-01-13 19:04:56 -07:00
Jaret Burkett 5b5aadadb8
Add LTX-2 Support (#644)
* WIP, adding support for LTX2

* Training on images working

* Fix loading comfy models

* Handle converting and deconverting lora so it matches original format

* Reworked ui to habdle ltx and propert dataset default overwriting.

* Update the way lokr saves to it is more compatable with comfy

* Audio loading and synchronization/resampling is working

* Add audio to training. Does it work? Maybe, still testing.

* Fixed fps default issue for sound

* Have ui set fps for accurate audio mapping on ltx

* Added audio procession options to the ui for ltx

* Clean up requirements
2026-01-13 04:55:30 -07:00
Jaret Burkett 926097aa4c Added 3bit ARA for qwen image 2512 2026-01-01 06:51:49 -07:00