 Jaret Burkett
|
ea47c120aa
|
Fix issue of saving full models when block compile is on
|
2026-10-23 12:33:49 -06:00 |
|
|
Jaret Burkett
|
1beba8a96d
|
Pass sample width/height/num_frames into cached sample prompt configs so reference sizing matches generation
|
2026-10-23 12:20:39 -06:00 |
|
|
Jaret Burkett
|
7861f92b13
|
Remove the forced no grad on vae decode for moss so it can be used with DFEs
|
2026-10-23 04:40:42 -06:00 |
|
|
Jaret Burkett
|
7cc9fef469
|
Add support for Bria Fibo 2.0
|
2026-10-22 18:11:17 -06:00 |
|
|
Jaret Burkett
|
14f8f95090
|
Bump node js version for docker and installer
|
2026-10-21 21:17:12 -06:00 |
|
|
Jaret Burkett
|
b1d5ef8534
|
Add support to train Fast H3 Omni with training adapter
|
2026-10-21 11:57:06 -06:00 |
|
|
Jaret Burkett
|
15303bf876
|
Moved x0 logic to the model class instead of recalculating it multiple times in the trainer
|
2026-10-21 11:38:09 -06:00 |
|
|
Jaret Burkett
|
c22f21fa95
|
Fix potential rotation mismatch issue and add webp support for the inactive fast image size checker
|
2026-10-21 07:00:36 -06:00 |
|
|
Jaret Burkett
|
04733873cf
|
Improve performance of encoding Moss prompts with a batch size > 1
|
2026-10-19 10:15:45 -06:00 |
|
|
Jaret Burkett
|
2d4c89d392
|
Fix issue with flash linear attention in docker image
|
2026-10-18 20:12:28 -06:00 |
|
|
Jaret Burkett
|
31f4d92f88
|
Version Bump
|
2026-10-18 13:46:14 -06:00 |
|
|
Jaret Burkett
|
4370d60904
|
Remove the compile for qwen3.8 since the kernels are optimized already
|
2026-10-18 13:30:04 -06:00 |
|
|
Jaret Burkett
|
85d01c6a04
|
Prevent possible traversal upstream through caption tags on captioner
|
2026-10-18 13:05:41 -06:00 |
|
|
Jaret Burkett
|
7c6b168c2c
|
Add tokens that will pull files with extensions git status
|
2026-10-18 13:04:58 -06:00 |
|
|
Jaret Burkett
|
e72468b456
|
Added initial support for fine tuning and captioning with Qwen 3.8 27B
|
2026-10-18 12:25:28 -06:00 |
|
|
Jaret Burkett
|
8cf43dd2cc
|
Fix issue that left text encoder on gpu after caching text embeddings.
|
2026-10-17 17:22:42 -06:00 |
|
|
Jaret Burkett
|
7b94dc5d8d
|
Add info about the Comfy UI node for the MOSS image edit loras
|
2026-10-17 15:46:40 -06:00 |
|
|
Jaret Burkett
|
577004a9cd
|
Publish libdevice as a module global so older triton builds without closure capture can compile the convrot kernels
|
2026-10-17 14:54:32 -06:00 |
|
|
Jaret Burkett
|
9d408eb64d
|
Fixed issue with Flux.3 controls where references were not scaled properly and more than one reference could be ignored in certain instances
|
2026-10-17 14:43:02 -06:00 |
|
|
Jaret Burkett
|
88360c68b2
|
Speed up MOSS-Image edit kv_cache training: reference isolation as unmasked flash attention instead of a boolean mask
|
2026-10-17 12:10:31 -06:00 |
|
|
Jaret Burkett
|
9a1828f201
|
Save qkv lora weights for MOSS Image as unfused LoRA, so they are more compatable with both fused and unfused weights.
|
2026-10-17 09:55:04 -06:00 |
|
|
Jaret Burkett
|
82e6311ce7
|
Add MOSS-Image SFT edit training
|
2026-10-16 17:07:13 -06:00 |
|
|
Jaret Burkett
|
dcebc7442d
|
Version Bump
|
2026-10-16 12:19:35 -06:00 |
|
|
Jaret Burkett
|
b7cd8a13af
|
Add support to train MOSS Image Base as an edit model
|
2026-10-16 12:19:06 -06:00 |
|
|
Jaret Burkett
|
0d19b26481
|
Version bump
|
2026-10-16 09:55:06 -06:00 |
|
|
Jaret Burkett
|
c6a9b56ae9
|
Add support for Moss Image Base
|
2026-10-16 09:54:38 -06:00 |
|
|
Jaret Burkett
|
80cb872a84
|
Fall back to an online load when a cached tokenizer/processor loads incomplete
|
2026-10-16 09:38:30 -06:00 |
|
|
Jaret Burkett
|
29f5ff452b
|
Fix MOSS-Image tokenizer loading on a fresh HF cache (missing chat template)
|
2026-10-16 09:33:05 -06:00 |
|
|
Jaret Burkett
|
e05a3a4f2c
|
Add support for MOSS Image
|
2026-10-16 08:09:45 -06:00 |
|
|
Jaret Burkett
|
12395a103f
|
Load comfy files if they exist in flux.3 image
|
2026-10-15 11:18:14 -06:00 |
|
|
Jaret Burkett
|
438595b78b
|
Fixed issue with caching text embeddings on Flux.3 Image
|
2026-10-15 09:56:40 -06:00 |
|
|
Jaret Burkett
|
0947356c62
|
Fix issue with loading flux3 text encoders in comfy format
|
2026-10-15 09:42:15 -06:00 |
|
|
Jaret Burkett
|
ef75577394
|
Version bump
|
2026-10-15 09:23:26 -06:00 |
|
|
Jaret Burkett (Ostris)
|
91bdc6dd15
|
Merge pull request #1078 from ostris/flux3-image
Add Support for FLUX.3 Image Klein Base
|
2026-10-15 09:18:34 -06:00 |
|
|
Jaret Burkett
|
9390ca463e
|
Merge branch 'main' into flux3-image
|
2026-10-15 08:26:39 -06:00 |
|
|
Jaret Burkett
|
831ccfa7aa
|
Added experimental band loss
|
2026-10-15 08:26:12 -06:00 |
|
|
Jaret Burkett
|
4f5b5cc7ad
|
flux3: store DiT and VAE in the comfy layout with renamed downloads
|
2026-10-14 14:06:19 -06:00 |
|
|
Jaret Burkett
|
3530876838
|
Merge branch 'main' into flux3-image
|
2026-10-14 13:41:20 -06:00 |
|
|
Jaret Burkett
|
7045370346
|
Keep cached pixel tensors as uint8 in the dataloader workers and only materialize to fp16 on the main thread
|
2026-10-14 13:40:49 -06:00 |
|
|
Jaret Burkett
|
dbbb7ee8f4
|
flux3: multi-reference sampling, match_target_res reference sizing, UI sections
|
2026-10-14 13:37:40 -06:00 |
|
|
Jaret Burkett
|
d4ffbcad70
|
flux3: drop dead gate_type path, reject flow-map packages, note refreshed base checkpoint
|
2026-10-14 13:03:00 -06:00 |
|
|
Jaret Burkett
|
7a64fdafad
|
Merge in from main
|
2026-10-14 11:27:16 -06:00 |
|
|
Jaret Burkett
|
f7a1fb9a77
|
Version bump
|
2026-10-09 21:09:34 -06:00 |
|
|
Jaret Burkett
|
f370e7441d
|
Add support for Qwen Image 2 Turbo with training adapter
|
2026-10-09 21:09:09 -06:00 |
|
|
Jaret Burkett
|
2fde764876
|
Version bump
|
2026-10-09 05:14:08 -06:00 |
|
|
Jaret Burkett
|
2fa0bd87e6
|
Add support for Iris-3B
|
2026-10-09 05:13:39 -06:00 |
|
|
Jaret Burkett
|
cb51e9a511
|
Add flux3 image
|
2026-10-09 03:08:46 -06:00 |
|
|
Jaret Burkett
|
a949f21f8e
|
Add ability to merge dataset buckets. Useful when using numerous different datasets.
|
2026-10-08 14:04:30 -06:00 |
|
|
Jaret Burkett
|
0cbac2c1f3
|
Add threadding to bucket resizing of a batch. Helpful to speed up large batch gathering.
|
2026-10-08 11:26:48 -06:00 |
|
|
Jaret Burkett
|
ecee894ed2
|
Key table rows by id so deleted rows don't leak pending state to neighbors
|
2026-09-27 10:19:02 -06:00 |
|
|
Jaret Burkett
|
6468a2ffa7
|
Fix issue loading new comfy checkpoints with ming
|
2026-09-26 13:32:05 -06:00 |
|
|
Jaret Burkett
|
16ae8d2999
|
Fixed issue on DGX spark that caused unnessary moves of models causing slow down on unified memory
|
2026-09-26 11:56:42 -06:00 |
|
|
Jaret Burkett
|
60d0c28c4d
|
Switch to new training adapters for all minimax models that no longer need contrastive guidance
|
2026-09-24 18:48:04 -06:00 |
|
|
Jaret Burkett
|
a776394353
|
Add support to train on Fast H3 v2 8 step directly with training adapter
|
2026-09-24 18:40:17 -06:00 |
|
|
Jaret Burkett
|
5aa3d0c9a8
|
Update ui to show DGX spark gpu usage correctly
|
2026-09-24 17:55:11 -06:00 |
|
|
Jaret Burkett
|
60ca869b38
|
Add proper support for DGX spark with easy manager. Thank you NVIDIA!
|
2026-09-24 17:44:40 -06:00 |
|
|
Jaret Burkett
|
cbc6bbdd70
|
Update ming image repo to new one
|
2026-09-24 08:40:16 -06:00 |
|
|
Jaret Burkett
|
460c29ba9c
|
Handle vision tokens in ming
|
2026-09-23 12:18:29 -06:00 |
|
|
Jaret Burkett
|
77847d7ff8
|
Add support for Ming-Image-0.1-Design with training adapter
|
2026-09-22 21:25:31 -06:00 |
|
|
Jaret Burkett
|
0bd34111b8
|
Version Bump
|
2026-09-22 14:26:15 -06:00 |
|
|
Jaret Burkett
|
07abdbedc7
|
Add match_target_res for Qwen Image 2.1 so reference images are scaled to the target's pixel area with their own aspect ratio, consistently across text-embedding caching, training, and sampling, and key the embedding cache on it.
|
2026-09-22 13:57:49 -06:00 |
|
|
Jaret Burkett
|
71bc877833
|
Fix support link
|
2026-09-22 12:52:29 -06:00 |
|
|
Jaret Burkett (Ostris)
|
8f2777a4f0
|
Merge pull request #1055 from vdruts/yue2-safetensors-heads
YuE2: support the safetensors-only v5/v8/v9 semantic heads, key latent cache by head
|
2026-09-22 12:45:25 -06:00 |
|
|
Jaret Burkett
|
f073188300
|
Fix Qwen Image Edit Plus batch_size > 1 without cached text embeddings by splitting the batch-stacked control tensor into one control image per prompt.
|
2026-09-22 12:14:19 -06:00 |
|
|
Jaret Burkett
|
a35d833abf
|
Fix text encoder unload skipping models whose pipeline does not expose the text encoder (Qwen-Image-2.1, Z-Image, LTX2), so cached-embedding runs actually free the TE.
|
2026-09-22 12:08:29 -06:00 |
|
|
Jaret Burkett
|
543a0d74d0
|
Fix issue with saving device state when a text encoder is unloaded.
|
2026-09-22 12:03:09 -06:00 |
|
|
Jaret Burkett (Ostris)
|
ae6ab90786
|
Merge pull request #1036 from ArcGleam/fix/qwen-edit-plus-batch-gt-1
Fix: QwenImageEditPlusModel crashes with batch_size > 1
|
2026-09-22 11:52:33 -06:00 |
|
|
Jaret Burkett (Ostris)
|
9b67238f6b
|
Merge pull request #1035 from ArcGleam/fix/qwen-edit-plus-control-image-inplace-downsample
Fix in-place downsampling of reference images in QwenImageEditPlusModel
|
2026-09-22 11:48:08 -06:00 |
|
|
Jaret Burkett (Ostris)
|
3a39d8cfc8
|
Merge pull request #1021 from hitoshic1982/fix/cache-text-encoder-device-preset
Fix cache_text_encoder device preset missing in BaseModel
|
2026-09-22 11:46:24 -06:00 |
|
|
Jaret Burkett
|
11b65a39e5
|
Ignore issue review folder
|
2026-09-22 11:05:38 -06:00 |
|
|
Jaret Burkett
|
a8dfcf7d7e
|
Save thumbnail with alpha if there is one.
|
2026-09-20 15:16:15 -06:00 |
|
|
Jaret Burkett
|
086b663bba
|
Add support for training with an alpha channel for Qwen Image 2.1
|
2026-09-20 14:49:53 -06:00 |
|
|
Jaret Burkett
|
31ddc709c3
|
Version bump
|
2026-09-20 11:04:32 -06:00 |
|
|
Jaret Burkett
|
474948abb4
|
Change default guidance to 3 for qwen image.
|
2026-09-20 11:04:16 -06:00 |
|
|
Jaret Burkett
|
f4f5d5dd58
|
Change qwen image 2 to use new lokr format
|
2026-09-20 11:02:57 -06:00 |
|
|
Jaret Burkett
|
c2622ed250
|
Add support for Qwen Image 2.1
|
2026-09-20 09:50:57 -06:00 |
|
|
Jaret Burkett
|
8fa15e3569
|
Move the ui model cards to their own plugin folders so plugins can add custom models in the trainer/generator/ui
|
2026-09-19 12:07:07 -06:00 |
|
|
Jaret Burkett
|
a6e3b18e8a
|
Added a notes section for jobs
|
2026-09-19 10:28:29 -06:00 |
|
|
Jaret Burkett
|
cc98c3b690
|
Add ability to train with just music or vocals on yue2. Doesnt work as expected though
|
2026-09-17 20:19:32 -06:00 |
|
|
Jaret Burkett
|
a9e7752178
|
Add MOSS music audio captioner
|
2026-09-17 10:45:37 -06:00 |
|
|
Jaret Burkett
|
080873bec9
|
Harden ace step transcription engine to detect and break out of loops. Add melbanroformer to seperate out the vocals and music for better transcriptions. Much better results.
|
2026-09-17 09:49:19 -06:00 |
|
|
Jaret Burkett
|
1db6083645
|
Show more information about LoRAs when loading loras in generate page
|
2026-09-17 09:35:43 -06:00 |
|
|
Jaret Burkett
|
ef8110cef7
|
Add LoRA uploader to the lora picker on the generator
|
2026-09-17 05:17:34 -06:00 |
|
|
Jaret Burkett
|
e65c4d0fb6
|
Fix issue with Krea2 patch size when training with shift
|
2026-09-16 13:46:39 -06:00 |
|
|
Jaret Burkett
|
410206e41c
|
Improve speed of sage sheet decoder with yue2. Recover from tracks that fail
|
2026-09-16 12:24:39 -06:00 |
|
|
Jaret Burkett
|
ce64e65c39
|
Add support for Qwen 2.5 Omni and add lora loading for captioning models
|
2026-09-16 10:36:19 -06:00 |
|
|
Jaret Burkett
|
2977672e4b
|
Point omni vllm to real repo
|
2026-09-16 09:37:21 -06:00 |
|
|
Jaret Burkett
|
db8dbd68c2
|
Add support for training VLLMs, omni models, and LLMs
|
2026-09-15 17:47:02 -06:00 |
|
|
Jaret Burkett
|
426ccb4ec4
|
fix file identifier parsing of some extensions that were not being recofnized in the ui
|
2026-09-15 12:59:54 -06:00 |
|
|
Jaret Burkett
|
6b2526a4d7
|
Allow Ace Step captioner to work with different output formats like yue2
|
2026-09-15 12:48:15 -06:00 |
|
|
Jaret Burkett
|
8bf12e476e
|
Fixed issue that could allow deleting and uploading items outside the restricted directories.
|
2026-09-15 10:13:53 -06:00 |
|
|
Jaret Burkett
|
07de70940a
|
Fix and optimize Ace Step captioner
|
2026-09-15 06:49:06 -06:00 |
|
|
Jaret Burkett
|
59c7de0539
|
Increase the repition penalty
|
2026-09-15 06:12:20 -06:00 |
|
|
Jaret Burkett
|
f2a2a6bea0
|
Make audio files in dataset be thumbnails until clicked and opened inn the viewer
|
2026-09-15 05:32:13 -06:00 |
|
|
Jaret Burkett (Ostris)
|
d673cc5205
|
Merge pull request #1042 from ostris/yue2
YuE2
|
2026-09-14 17:55:47 -06:00 |
|
|
Jaret Burkett
|
48d6f441cc
|
Version bump
|
2026-09-14 17:50:36 -06:00 |
|
|
Jaret Burkett
|
213d32cad3
|
Train 50% of the time with and without sheet music.
|
2026-09-14 17:50:15 -06:00 |
|
|
Jaret Burkett
|
f9b23b40bb
|
Load sage sheet and use it on training with 50% dropout so it generates sheet music and handles without it.
|
2026-09-14 17:01:36 -06:00 |
|
|
Jaret Burkett
|
9ea4348d1e
|
Normalize captions
|
2026-09-14 14:51:14 -06:00 |
|
|
Jaret Burkett
|
5ae240dd81
|
Update readme
|
2026-09-14 14:18:45 -06:00 |
|