Jaret Burkett
171535833a
Add Mac OS support for Apple Silicon ( #770 )
...
* Made an install script and auto updates env for mac
* GPU sensors and initial training working for MAC. Still WIP.
* Switch dataloader to single threaded until I can work around some mac pickeling issues.
* Get quantization working on mac
* Fix mac exclusive imports so they don't break other builds.
* Add mac instructions to the UI
2026-03-30 09:37:47 -06:00
Jaret Burkett
fbda10d088
Add a duplicate dataset function to the ui
2026-03-29 13:51:18 -06:00
Jaret Burkett
45e99664b9
Add icons to the top bar on the job page
2026-03-29 12:38:47 -06:00
Jaret Burkett
540659709d
Improved the load time of dataset and sample images and videos by switching to streaming
2026-03-29 10:38:34 -06:00
Jaret Burkett
e030f4f2e0
Show the control images in the image viewer when clicked so they can be easily previewed for reference.
2026-03-29 10:00:54 -06:00
willhsmit
20929b93df
Fix onChange path for EMA Decay input ( #695 )
...
Changes to the EMA Decay input don't get preserved when switching back and forth between Advanced and Simple view. I believe the onChange is not writing it correctly here.
2026-03-28 13:02:32 -06:00
Rayane
ca72eb1515
Add 1328 native resolution for Qwen Image training ( #749 )
...
* Add 1328 native resolution for Qwen Image training
Qwen-Image and Qwen-Image-2512 have a native 1:1 resolution of 1328x1328
as documented in the official model card's aspect ratio table. Adding it
to the resolution buckets and UI allows training at the model's native
resolution for improved quality.
* Revert example config change (24GB OOM at 1328)
2026-03-28 12:09:15 -06:00
Jaret Burkett
740657e25e
Improve dataset uploader. Upload the files one at a time instead of one huge chunk. Show progress for each file.
2026-03-27 09:26:22 -06:00
Jaret Burkett
2c8d2acdcb
On jobs table, sort idle jobs by last updated so recent active ones are at the top
2026-03-26 10:33:17 -06:00
Jaret Burkett
9a77389653
On a new training job, or when editing one, load everything before allowing editing
2026-03-26 10:23:42 -06:00
Jaret Burkett
a7bb4ddb2c
Work on loss graph. Add smoothed overlay. Allow user to hilite a secton of the graph to zoom into.
2026-03-26 10:05:09 -06:00
Jaret Burkett
4df3b0463f
Save job pid to the database and sing sigint to kill it when stopping so it stops immediatly.
2026-03-26 09:10:37 -06:00
Jaret Burkett
89d2090962
Fixed race condition that would occasionally set the dataset path to the first one when editing a job
2026-03-25 11:22:42 -06:00
Jaret Burkett
5d5a8ef9da
Fixed issue with deleting datasets and jobs with newer version of node.js. Bumped minimum version of node js to 20
2026-03-25 10:04:36 -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
Rasmus Lerdorf
4ad14d211a
Add an import config button ( #733 )
2026-03-23 15:41:27 -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
35978df8a3
Adjust defaults for ui graph to get and show all losses
2026-03-02 10:27:15 -07:00
Jaret Burkett
57d407cfd4
Add support for training lodestones/Zeta-Chroma
2026-03-01 12:52:29 -07:00
Jaret Burkett
1c74ca5d22
Add audio_loss_multiplier to scale audio loss to larger values if desired.
2026-02-19 11:57:44 -07:00
Jaret Burkett
42acb0d4be
Build out an audio player card in preperation for audio datasets and samples.
2026-02-03 08:15:55 -07:00
Jaret Burkett
ea912d2d7b
Increase default sample steps from 25 to 30 for z_image
2026-01-27 09:39:21 -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
e132dbae76
Add number of repeats for a dataset in the ui
2026-01-15 08:03:31 -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
6870ab490f
Add 4 bit ARA for qwen image 2512
2026-01-01 20:16:56 -07:00
Jaret Burkett
926097aa4c
Added 3bit ARA for qwen image 2512
2026-01-01 06:51:49 -07:00
Jaret Burkett
4d5a649a7d
Added initial support for Qwen-Image-2512
2025-12-31 06:11:56 -07:00
Jaret Burkett
8edf1e44c5
Added 3 bit accuracy recovery adapter for qwen image edit 2511
2025-12-24 05:33:47 -07:00
Jaret Burkett
57a2ab1299
Add initial support for Qwen Image Edit 2511
2025-12-23 10:53:48 -07:00
Jaret Burkett
9883055684
Fix issue where ui could break if caption is read as a non string.
2025-12-23 09:47:19 -07:00
Jaret Burkett
ba00eea7d9
Add loss graph to the ui
2025-12-18 10:08:59 -07:00
Jaret Burkett
26f4f02453
Add support for Z-Image-De-Turbo
2025-12-04 10:03:13 -07:00
Jaret Burkett
6c85184441
Set zit training adapter to default to v2
2025-12-02 16:46:05 -07:00
Jaret Burkett
4e62c38df5
Add support for training Z-Image Turbo with a de-distill training adapter
2025-11-28 08:08:53 -07:00
Jaret Burkett
01cf480233
Add FLUX.2 official weights
2025-11-25 08:52:19 -07:00
Jaret Burkett
dadbeda197
Update test weights
2025-11-23 10:51:50 -07:00
Jaret Burkett
0b5f3475e2
Merge branch 'main' into flux2
2025-11-23 08:18:35 -07:00
Jaret Burkett
26e4b71b57
Fix issue with parsing image info for sample info on some windows machines.
2025-11-19 08:45:05 -07:00
Jaret Burkett
cd607c4902
Fix a bug that can happen if you remove a gpu from your machine.
2025-11-19 08:28:48 -07:00
Jaret Burkett
af8e9ea149
Add initial support for FLUX.2
2025-11-18 11:17:38 -07:00
Jaret Burkett
2e7b2d9926
Added Differential Guidance training target
2025-11-10 09:38:25 -07:00
Jaret Burkett
ee206cfa18
Added blank prompt preservation
2025-10-22 14:55:13 -06:00
Jaret Burkett
ca57ffc270
When having less than 3 sample images, add spacing to the grid so images are not huge
2025-10-22 13:52:38 -06:00
Jaret Burkett
76ce757e0c
Added initial support for layer offloading wit Wan 2.2 14B models.
2025-10-20 14:54:30 -06:00
Jaret Burkett
645c54d617
Fixed issue that may occour if no queue is built when starting one from the table.
2025-10-15 08:48:07 -06:00
Jaret Burkett
b523d58699
Added ability to clone an existing job in the ui
2025-10-14 14:13:37 -06:00
Jaret Burkett
7e34a03113
Added queing system to the UI
2025-10-14 12:00:42 -06:00