Commit Graph
560 Commits
Author SHA1 Message Date
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 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 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 7b94dc5d8d Add info about the Comfy UI node for the MOSS image edit loras 2026-10-17 15:46:40 -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 b7cd8a13af Add support to train MOSS Image Base as an edit model 2026-10-16 12:19:06 -06:00
Jaret Burkett c6a9b56ae9 Add support for Moss Image Base 2026-10-16 09:54:38 -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 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 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 f370e7441d Add support for Qwen Image 2 Turbo with training adapter 2026-10-09 21:09:09 -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 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 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 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 (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 (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
vdruts d0ecde7779 YuE2: load safetensors semantic heads / NAR adapters (v5, v8, v9) and key the latent cache by head 2026-09-21 09:21:44 -04: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 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 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