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
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
0b9c365acb
Add flac and ogg support
2026-04-10 12:10:54 -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
853ffaf207
Add light mode support.
2026-03-31 16:54:55 -06:00
e22039e4aa
Add more optimizers to the ui
2026-03-31 09:20:20 -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
ac82ebd852
Dont try to list hidden files in datasets
2026-03-30 10:03:10 -06:00
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
fbda10d088
Add a duplicate dataset function to the ui
2026-03-29 13:51:18 -06:00
45e99664b9
Add icons to the top bar on the job page
2026-03-29 12:38:47 -06:00
540659709d
Improved the load time of dataset and sample images and videos by switching to streaming
2026-03-29 10:38:34 -06:00
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
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
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
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
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
9a77389653
On a new training job, or when editing one, load everything before allowing editing
2026-03-26 10:23:42 -06:00
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
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
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
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
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
4ad14d211a
Add an import config button ( #733 )
2026-03-23 15:41:27 -06:00
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
35978df8a3
Adjust defaults for ui graph to get and show all losses
2026-03-02 10:27:15 -07:00
57d407cfd4
Add support for training lodestones/Zeta-Chroma
2026-03-01 12:52:29 -07:00
1c74ca5d22
Add audio_loss_multiplier to scale audio loss to larger values if desired.
2026-02-19 11:57:44 -07:00
42acb0d4be
Build out an audio player card in preperation for audio datasets and samples.
2026-02-03 08:15:55 -07:00
ea912d2d7b
Increase default sample steps from 25 to 30 for z_image
2026-01-27 09:39:21 -07:00
2db090144a
Add support for Z-Image
2026-01-27 09:34:46 -07:00
9ef6f1a828
Increase client body size to 100 gb
2026-01-24 12:44:17 -07:00
a6da9e37ac
Add support for FLUX.2 klein base models
2026-01-17 17:46:25 -07:00
e132dbae76
Add number of repeats for a dataset in the ui
2026-01-15 08:03:31 -07:00
64fe29b182
Support img 2 vid training for ltx-2
2026-01-13 19:04:56 -07:00
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
6870ab490f
Add 4 bit ARA for qwen image 2512
2026-01-01 20:16:56 -07:00
926097aa4c
Added 3bit ARA for qwen image 2512
2026-01-01 06:51:49 -07:00
4d5a649a7d
Added initial support for Qwen-Image-2512
2025-12-31 06:11:56 -07:00
8edf1e44c5
Added 3 bit accuracy recovery adapter for qwen image edit 2511
2025-12-24 05:33:47 -07:00
57a2ab1299
Add initial support for Qwen Image Edit 2511
2025-12-23 10:53:48 -07:00
9883055684
Fix issue where ui could break if caption is read as a non string.
2025-12-23 09:47:19 -07:00
113bbd0e3e
Fixed an issue with the new version of nextjs with client body size
2025-12-18 19:18:35 -07:00
ba00eea7d9
Add loss graph to the ui
2025-12-18 10:08:59 -07:00
cd0e691040
Fix NextJS vulnerability ( #594 )
...
* Fix NextJS vulnerability
https://nextjs.org/blog/CVE-2025-66478
* Update package-lock.json
* Update package-lock.json
* Update package.json
* Update package-lock.json
2025-12-17 10:03:13 -07:00
26f4f02453
Add support for Z-Image-De-Turbo
2025-12-04 10:03:13 -07:00
6c85184441
Set zit training adapter to default to v2
2025-12-02 16:46:05 -07:00
4e62c38df5
Add support for training Z-Image Turbo with a de-distill training adapter
2025-11-28 08:08:53 -07:00
01cf480233
Add FLUX.2 official weights
2025-11-25 08:52:19 -07:00
dadbeda197
Update test weights
2025-11-23 10:51:50 -07:00
0b5f3475e2
Merge branch 'main' into flux2
2025-11-23 08:18:35 -07:00
26e4b71b57
Fix issue with parsing image info for sample info on some windows machines.
2025-11-19 08:45:05 -07:00
cd607c4902
Fix a bug that can happen if you remove a gpu from your machine.
2025-11-19 08:28:48 -07:00
af8e9ea149
Add initial support for FLUX.2
2025-11-18 11:17:38 -07:00
2e7b2d9926
Added Differential Guidance training target
2025-11-10 09:38:25 -07:00
ee206cfa18
Added blank prompt preservation
2025-10-22 14:55:13 -06:00
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
76ce757e0c
Added initial support for layer offloading wit Wan 2.2 14B models.
2025-10-20 14:54:30 -06:00
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
b523d58699
Added ability to clone an existing job in the ui
2025-10-14 14:13:37 -06:00
7e34a03113
Added queing system to the UI
2025-10-14 12:00:42 -06:00
e9c4d94256
Allow for matching target resolution with control images for Qwen Image Edit 2509
2025-10-10 14:24:27 -06:00
1bc6dee127
Change auto_memory to be layer_offloading and allow you to set the amount to unload
2025-10-10 13:12:32 -06:00
dfc85f0b51
Add Auto Memory for qwen models in the ui
2025-10-07 10:43:52 -06:00
1ea50d8590
Add cpu info the the job page
2025-10-07 08:30:23 -06:00
2e9de5eb50
Add ability to delete samples from the ui
2025-09-29 04:49:32 -06:00
ebadb321e3
On samples page, auto scroll to bottom on load. Added a floating button to scroll to bottom.
2025-09-29 03:56:17 -06:00
c233a80337
Reqorked visibility toggle on samples, should help when dealing with more samples
2025-09-28 14:13:10 -06:00
c20240be82
Add advanced menu on job to allow user to do things like make a job as stopped if the status ever gets hung
2025-09-28 13:43:00 -06:00
4e207d92cd
Add seed to the sample image modal
2025-09-28 12:54:34 -06:00
f0646a0a70
Reworked ui sample image modal to show more information and function a lot better.
2025-09-27 12:50:47 -06:00
98d35f36a9
Add hidream ARA
2025-09-27 09:31:23 -06:00
1069dee0e4
Added ui sopport for multi control samples and datasets. Added qwen image edit 5209 to the ui
2025-09-25 11:10:02 -06:00
f74475161e
Add stepped loss type
2025-09-22 15:50:12 -06:00
2120dc5936
Upgrade job to new ui trainer to fix issue with slider config showing up on old configs.
2025-09-17 13:41:48 -06:00
218f673e3d
Added support for new concept slider training script to CLI and UI
2025-09-16 10:22:34 -06:00
0f2239ca23
Add force sample toggle to the ui
2025-08-31 16:58:27 -06:00
fd13bd73a6
Add a Download button on samples to download all the samples as a zip file
2025-08-27 09:12:46 -06:00
5ad190b11d
Improve UI for sample images when there are no samples
2025-08-27 08:10:24 -06:00
d0338b8b0b
Allow dropping images directly into dataset folder without having to open the add images modal. Improve ui flow of dataset messaging.
2025-08-25 14:04:48 -06:00
37eda7b2e2
Add a tab to the UI to show the config file for the job. Read only.
2025-08-25 13:08:40 -06:00
24372b5e35
Add toggles to the UI to add flipped versions of the datasets, X, Y or both.
2025-08-24 13:39:04 -06:00
823e690703
Changed auth to use the wording 'password' instead of 'token' and give information about defaults and how to change the password.
2025-08-23 09:30:44 -06:00
e1fd411665
Added support for Chroma1 official release. Will still use single file verstion instead of the diffusers version.
2025-08-23 09:06:28 -06:00
59ff4efae5
Add support for training Qwen Image Edit in the UI
2025-08-22 10:26:46 -06:00
38d3814be7
Added 4bit ARAs for Wan 2.2 14b models
2025-08-21 08:16:07 -06:00
d2bbe1872c
Add support for fine tuning Wan 2.2 I2V 14B
2025-08-18 11:43:32 -06:00
8ea2cf00f6
Added training to the ui. Still testing, but everything seems to be working.
2025-08-16 05:51:37 -06:00
ca7bfa414b
Increase max number of samples to 40
2025-08-16 05:27:38 -06:00
77b10d884d
Add support for training with an accuracy recovery adapter with qwen image
2025-08-12 08:21:36 -06:00
bb6db3d635
Added support for caching text embeddings. This is just initial support and will probably fail for some models. Still needs to be ompimized
2025-08-07 10:27:55 -06:00
5d8922fca2
Add ability to designate a dataset as i2v or t2v for models that support it
2025-08-06 09:29:47 -06:00
23327d5659
Add qwen image to the ui
2025-08-04 15:48:51 -06:00
ef1d60ba34
Update wan 2.2 5b timestep distribution to weighted.
2025-07-30 10:13:22 -06:00
ca7c5c950b
Add support for Wan2.2 5B
2025-07-29 05:31:54 -06:00
e55116d8c9
Added hidream low vram options
2025-07-27 18:29:46 -06:00
99705ec8be
Add support in UI for Hidream E1
2025-07-27 18:13:36 -06:00
ed8d14225f
Add ability to set the quantization type for text encoders and transformer in the ui
2025-07-27 18:00:53 -06:00
3e14a674ac
Fix upload progress for datasets in the ui
2025-07-26 09:07:30 -06:00
523c159579
Add vram flag to some models in the ui
2025-07-24 07:02:46 -06:00
8610c6ed7f
Made it easy to add control images to the samples in the UI
2025-07-17 12:00:48 -06:00
ef5149180c
Switch i2v ui defaults to weighted
2025-07-12 21:30:04 -06:00
7ab44ae0cd
Fix issue with getting captions on runpod
2025-07-11 19:16:50 +00:00
47002b067f
Add path to image for datasets on the image card
2025-07-11 11:44:48 -06:00
8537a8557f
Add simple ui settings to train Wan i2v models.
2025-07-11 11:28:40 -06:00
611969ec1f
Allow control image for omnigen training and sampling
2025-07-09 13:54:55 -06:00
60ef2f1df7
Added support for FLUX.1-Kontext-dev
2025-06-26 15:24:37 -06:00
19ea8ecc38
Added support for finetuning OmniGen2.
2025-06-25 13:58:16 -06:00
4586eb5392
Added social links to sidebar
2025-06-17 13:25:24 -06:00
ff617fdaea
Started doing info bubble docs on the simple ui
2025-06-17 11:00:24 -06:00
595a6f1735
Initial setup for a cron working on the ui for various tasks
2025-06-17 07:43:34 -06:00
11f2eee53a
Hide control images from ui image viewer
2025-06-16 07:18:43 -06:00
f8fb3b9c45
Added support for sdxl and sd1.5 to the ui.
2025-06-10 10:03:54 -06:00
d5c547da43
Fixed DOP typo
2025-06-10 08:44:47 -06:00
f19f7f9486
Fixed issue with wan2.1 training in ui. Name had a typo
2025-06-10 08:42:18 -06:00
7317ed58af
Adjust the ui of the sidebar
2025-06-10 08:40:51 -06:00
517bc294fa
Update support link
2025-06-10 08:28:30 -06:00
25e150b370
Added support for Flex.2 in the UI
2025-05-07 12:41:51 -06:00
43cb5603ad
Added chroma model to the ui. Added logic to easily pull latest, use local, or use a specific version of chroma. Allow ustom name or path in the ui for custom models
2025-05-07 12:06:30 -06:00
79c87701e7
Add hidream to the ui
2025-04-16 13:45:21 -06:00
b0d0466efd
Add better error messages if name exists when saving a job
2025-04-03 11:25:25 -06:00
b94d7aafea
Have error boundary if simple job cannot be displayed due to the job being advanced
2025-03-27 19:55:40 -06:00
e4526ad4a4
Updates to handle video in a dataset on ui
2025-03-26 12:15:28 -06:00
9a902c067f
Show the console log on the job overview in the ui.
2025-03-21 11:45:36 -06:00
0bbc69c135
Fix issue where the job would hang in the ui if it failed to start
2025-03-21 09:46:41 -06:00
c8049a483d
Fix issue with auth token not on build time
2025-03-20 17:27:30 -06:00
3a6b24f4c8
Added a way to secure the UI. Plus various bug fixes and quality of life updates
2025-03-20 08:07:09 -06:00
6cde96ae5f
Adding files I forgot to stage to the last commit
2025-03-15 11:59:41 -06:00
1be613ed06
Added advanced mode yaml editor to the ui
2025-03-15 11:58:59 -06:00
cf4216e6b8
Add support for wan training in ui
2025-03-13 18:54:27 -06:00
7ae31c9ae9
Added LoKr to the ui
2025-03-02 08:49:01 -07:00
f6e16e582a
Added Differential Output Preservation Loss to trainer and ui
2025-02-25 20:12:36 -07:00
259ded9602
Fixed issue with trigger word saving in ui
2025-02-24 11:04:24 -07:00
440ba5fb3d
Spawn windows in an cmd terminal. Should be working now, but not sure on my system
2025-02-24 08:54:56 -07:00
093f14ac19
UI Bug fixes and initial windows support
2025-02-24 08:15:22 -07:00
1d0e3a4498
Fixed some build issues for now. Added info to the readme
2025-02-23 15:59:17 -07:00
3c7daf49f3
Add HF token to env when spawing via ui
2025-02-23 14:52:12 -07:00
56d8d6bd81
Capture speed from the timer for the ui
2025-02-23 14:38:46 -07:00
3e49337a58
Set step to the last step saved at when exiting
2025-02-23 13:21:22 -07:00
b366e46f1c
Added more settings to the training config
2025-02-23 12:34:52 -07:00
a280f78c69
Added checkpoint downloader
2025-02-22 16:48:15 -07:00
6e19e7449e
Fixed some issues with gpu info refreshing
2025-02-22 14:14:23 -07:00
a6d46ad9ae
Cleanup of job page
2025-02-22 13:54:06 -07:00
f3725578dd
Cleaned up dashboard
2025-02-22 13:23:26 -07:00
ed99c3c0c8
Moved gpu to its own widget
2025-02-22 12:43:20 -07:00
Jaret Burkett (Ostris)
willhsmit
Rayane
Rasmus Lerdorf
apolinário