Jaret Burkett 609f3e5fa5 Fix issue with memory manager on MPS devices

2026-10-23 14:37:50 -06:00

..

anima

Move the offload and quantization to the base model

2026-08-28 08:37:56 -06:00

boogu_image

Move the offload and quantization to the base model

2026-08-28 08:37:56 -06:00

chroma

Performance fixes and bug fixes

2026-08-28 22:21:09 -06:00

ernie_image

Performance fixes and bug fixes

2026-08-28 22:21:09 -06:00

example_model

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

f_light

Performance fixes and bug fixes

2026-08-28 22:21:09 -06:00

fibo2

Add support for Bria Fibo 2.0

2026-10-22 18:11:17 -06:00

flux2

Performance fixes and bug fixes

2026-08-28 22:21:09 -06:00

flux3

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

flux_kontext

Performance fixes and bug fixes

2026-08-28 22:21:09 -06:00

hidream

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

ideogram4

Move the offload and quantization to the base model

2026-08-28 08:37:56 -06:00

iris3b

Add support for Iris-3B

2026-10-09 05:13:39 -06:00

krea2

Fix issue with memory manager on MPS devices

2026-10-23 14:37:50 -06:00

ltx2

Fix issue where we gave the wrong dtype hidden states

2026-09-03 08:43:15 -06:00

mageflow

Move the offload and quantization to the base model

2026-08-28 08:37:56 -06:00

ming_image

Update ming image repo to new one

2026-09-24 08:40:16 -06:00

minimax_h3

Add support to train Fast H3 Omni with training adapter

2026-10-21 11:57:06 -06:00

moss_image

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

nucleus_image

Performance fixes and bug fixes

2026-08-28 22:21:09 -06:00

omnigen2

Move the offload and quantization to the base model

2026-08-28 08:37:56 -06:00

prx_pixel_t2i

Move the offload and quantization to the base model

2026-08-28 08:37:56 -06:00

qwen_image

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

qwen_image_2

Add support for Qwen Image 2 Turbo with training adapter

2026-10-09 21:09:09 -06:00

wan22

Performance fixes and bug fixes

2026-08-28 22:21:09 -06:00

z_image

Stability improvements to model offloading. Added D-OPSD bleed loss as well.

2026-08-29 11:23:47 -06:00

zeta_chroma

Performance fixes and bug fixes

2026-08-28 22:21:09 -06:00

__init__.py

Add support for Bria Fibo 2.0

2026-10-22 18:11:17 -06:00

ui.tsx

Add support for Bria Fibo 2.0

2026-10-22 18:11:17 -06:00