Commit Graph

121 Commits

Author SHA1 Message Date
Jaret Burkett
0d018db689 Rework the toggles so we can hide the trend line from the loss graph. No need for a raw toggle either. 2026-06-08 13:56:37 -06:00
Jaret Burkett
687def6f7a Add better smoothing with ema rounding of the ends of the loss graph for first and last instance do not dominate the smoothing factors. 2026-06-08 10:43:46 -06:00
Jaret Burkett
d7f8887bbf Show all logged metrics on the loss graph 2026-06-08 09:43:03 -06:00
Jaret Burkett
41157b460c Added ability to set the caption extention in dataset viewer, captioner, and trainer so one dataset can have multiple caption styles in different files with different extensions. Added dataset caption template for a blank ideogram 4 formatted template. 2026-06-06 08:32:24 -06:00
Jaret Burkett
21a6beb194 Allow editing the boxes visually on the sample section for ideogram 2026-06-05 14:09:18 -06:00
Jaret Burkett
4441080c05 Rework the ui of ideogram dataset image caption editor 2026-06-05 13:20:58 -06:00
Jaret Burkett
6fd2253932 Imporved editing boxed on the dataset viewer 2026-06-05 10:30:06 -06:00
Jaret Burkett
ef12260b80 Add a prompt upsample ui for upsampling prompts to ideogram format prompts. 2026-06-05 09:56:02 -06:00
Jaret Burkett
90a2084f70 Allow adjusting, adding, and deleting bounding boxes. 2026-06-04 20:29:31 -06:00
Jaret Burkett
bb60f6d1d1 Show bounding boxes on the sample image card if we have them in the prompt 2026-06-04 15:59:02 -06:00
Jaret Burkett
edcc7415d1 Added an autocaptioner for Ideogram 4 captions 2026-06-04 13:04:23 -06:00
Jaret Burkett
b5f21ae695 Fixed issue where the training type dropdown in the ui was not showing fully 2026-06-03 10:14:15 -06:00
Jaret Burkett
bd468727a6 Fixed issue where some image cards in samples and datasets would show gray instead of the image until you scroll. 2026-06-01 03:36:32 -06:00
Jaret Burkett
212cfe998a Add ability to download or delete the optimizer state from the ui 2026-05-29 08:56:22 -06:00
Jaret Burkett
30886b8f92 Added a wallet indicator for Ostris Cloud 2026-05-28 11:09:51 -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
76c99da4e4 Add version to the ui 2026-05-26 07:30:09 -06:00
Jaret Burkett
266956068a Add a way to delete checkpoints on the ui 2026-05-26 07:09:21 -06:00
Jaret Burkett
954c5efec8 Show job info on sidebar active job in ui 2026-05-25 11:07:29 -06:00
Jaret Burkett
083236a2a7 Drastically improve the loading speed of images in the ui by using a custom loader and abort controller to abort when images leave the view. 2026-05-25 10:04:25 -06:00
Jaret Burkett
3d836ac371 Show active jobs in the sidebar of the ui 2026-05-25 08:21:42 -06:00
Jaret Burkett
a798e06dd2 Updated the support button to look more like a button 2026-05-25 07:48:04 -06:00
Jaret Burkett
8042cbe9d2 Added virtulization for sample images to handle huge number of samples more efficientyly. 2026-05-25 07:22:24 -06:00
Jaret Burkett
307ff11bc5 Drastically improved the performance of the dataset viewer on large datasets by switching to virtualization. Fixed issue with images loading when they have two periods in a row .. 2026-05-24 15:13:55 -06:00
Jaret Burkett
c6a7e81a70 Added a dataset image viewer. Fixed an issue where captions would show as not saved when they were saved. 2026-05-24 14:31:06 -06:00
Jaret Burkett
6fc08a8928 Fix issue with saving images from new sample viewer 2026-05-22 11:58:32 -06:00
Jaret Burkett
15d4fb89ff Fixed overflow issue on modal 2026-05-19 10:12:43 -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
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
0d91fcee9e Allow edit of captioning job 2026-04-30 16:00:16 -06:00
Jaret Burkett
43989cc19e Add advanced config section to the captioner 2026-04-28 10:49:46 -06:00
Jaret Burkett
f972b750e6 Performance improvements to captioner. Add ability to set a default caption for ace captioner to avoid needing to caption the audio so we only transcribe 2026-04-28 10:15:11 -06:00
Jaret Burkett
0b9c365acb Add flac and ogg support 2026-04-10 12:10:54 -06:00
Jaret Burkett
145144eee3 Fix issue with auto updating captions when captioning a dataset 2026-04-10 11:51:18 -06:00
Jaret Burkett
765a9d5b2e Add a download button to music samples in in the gear menu 2026-04-10 10:07:45 -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
b43ea6c2d3 Abort caption requests when they are not in view to tax the server less. 2026-03-30 10:46:25 -06:00
Jaret Burkett
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
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
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
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
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
35978df8a3 Adjust defaults for ui graph to get and show all losses 2026-03-02 10:27:15 -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
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
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