Commit Graph
1604 Commits
Author SHA1 Message Date
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
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 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