Commit Graph
227 Commits
Author SHA1 Message Date
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
Jaret Burkett 9113420b61 Phase 2 2026-08-27 15:37:13 -06:00
Jaret Burkett 9ed2e0b8e7 Test model loading 2026-08-27 14:31:53 -06:00
Jaret Burkett 8db198ec0a Phase 1 2026-08-27 11:53:08 -06:00
Jaret Burkett e8d9cf6d35 Models v2 - phase 0 2026-08-27 10:51:43 -06:00
Jaret Burkett da79ebce99 Add D-OPSD as a distillation handeling option for MiniMax H3 ref2va 2026-08-26 11:06:33 -06:00
Jaret Burkett 89102f76dc Allow user to pick if reference images are sent in the image or video reference stream for Hidream H3 ref2va 2026-08-19 13:32:06 -06:00
Jaret Burkett afd1d92722 Fix issue with double sampling qwen vl video frames for hidream h3 for reference videos 2026-08-17 15:39:07 -06:00
Jaret Burkett 42dfe9c661 Improve video frame reader 2026-08-16 10:21:11 -06:00
Jaret Burkett 2cbc2bb097 On minimax h3, trim cached text embeddings to max tokens if they are longer than specified 2026-08-16 08:42:01 -06:00
Jaret Burkett 151ad0e959 Adjust MiniMah h3 sizing for videos to downscale to match 2026-08-15 07:50:42 -06:00
Jaret Burkett 127d6f626d Rework img/video reference in Minimax H3 to more closely match the comfy ui implementation. 2026-08-15 07:13:54 -06:00
Jaret Burkett 97bf49edad Add support for video references in MiniMax H3 ref2va 2026-08-15 06:18:09 -06:00
Jaret Burkett 6ea281973d Add support for MiniMax H3 Ref2Vid training 2026-08-13 12:36:03 -06:00