Commit Graph

117 Commits

Author SHA1 Message Date
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
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
2e7b2d9926 Added Differential Guidance training target 2025-11-10 09:38:25 -07: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