Commit Graph
240 Commits
Author SHA1 Message Date
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 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 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
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 e65c4d0fb6 Fix issue with Krea2 patch size when training with shift 2026-09-16 13:46:39 -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 6e1b39c7a8 Improve ui and functionality 2026-09-10 15:44:00 -06:00
Jaret Burkett 449488f6b7 Initial inference engine build out 2026-09-10 14:22:49 -06:00
Jaret Burkett f56b5a1d40 Change hidream to be proper x0 prediction and use a custom weighing scheme for x0 models. 2026-09-10 11:49:54 -06:00
Dawn-Lcg b90cccdfb1 Fix: QwenImageEditPlusModel crashes with batch_size > 1
Pass only the current sample's caption to encode_prompt, and right-pad
per-sample embeddings before concatenating.
2026-09-10 20:31:30 +08:00
Dawn-Lcg 17e5f59f36 Fix in-place downsampling of reference images in QwenImageEditPlusModel
Copy control images to avoid mutating the original list.
2026-09-10 19:48:19 +08:00
Jaret Burkett 90515c5396 Fix issue where we gave the wrong dtype hidden states 2026-09-03 08:43:15 -06:00
Jaret Burkett 764b5064fb Migrate to a new DTO for latents to carry more information that a normal tensor such as audio. 2026-08-30 10:30:21 -06:00
Jaret Burkett 2a69c1e7de Add initial support for Minimax H3 VSA sparse attention 2026-08-30 08:24:43 -06:00
Jaret Burkett 683fe8afc0 Stability improvements to model offloading. Added D-OPSD bleed loss as well. 2026-08-29 11:23:47 -06:00
Jaret Burkett 7195abc32b Performance fixes and bug fixes 2026-08-28 22:21:09 -06:00
Jaret Burkett 92df289931 Fix quantization issues 2026-08-28 10:48:34 -06:00
Jaret Burkett 85a6880643 Move the offload and quantization to the base model 2026-08-28 08:37:56 -06:00
Jaret Burkett 702254688d Added legacy paths 2026-08-28 07:55:19 -06:00
Jaret Burkett c3bc8b0b4e Full test of ui arch compatability 2026-08-28 06:56:05 -06:00
Jaret Burkett 520d96aac3 Phase 2 2026-08-27 18:18:31 -06:00