Jaret Burkett
|
a4bbe167ce
|
Added reference token attention isolation (kv_cache) for Krea2 edit training. Same training cost with significant inference speed up. 2x inference speedup.
|
2026-07-09 12:00:55 -06:00 |
|
Jaret Burkett
|
6233efe1bb
|
Fix sampling issue on zimage turbo and krea2 turbo models
|
2026-07-08 12:34:39 -06:00 |
|
Jaret Burkett
|
7bceec3b07
|
Apply a loss mask for first frame conditioning for wan and ltx
|
2026-07-07 09:05:53 -06:00 |
|
Jaret Burkett
|
bd93a312bc
|
Load more types of comfy ui style quants for ltx2
|
2026-07-07 09:04:51 -06:00 |
|
Jaret Burkett
|
b1e1a834d4
|
Added support to train Krea2 as an edit model
|
2026-07-04 09:12:39 -06:00 |
|
Jaret Burkett
|
4e50535478
|
Allow models to provide an additional loss
|
2026-06-28 09:49:36 -06:00 |
|
Jaret Burkett
|
1e12b6b73f
|
Remove duplicate save when triggering save next step
|
2026-06-28 07:18:16 -06:00 |
|
Jaret Burkett
|
7ee1f98f6d
|
Allow loading and saving z_image in comfy safetensors format.
|
2026-06-28 07:15:49 -06:00 |
|
Jaret Burkett
|
f8667f0334
|
Save Krea2 even if quantized
|
2026-06-28 06:03:37 -06:00 |
|
Jaret Burkett
|
4a99ddabad
|
Fix breaking change with diffusers qwen image
|
2026-06-26 10:04:35 -06:00 |
|
Jaret Burkett
|
5f04ae7ad5
|
Rework merge_network_on_save to handle dequantization on merging and saving for much more efficient full finetuning.
|
2026-06-25 13:19:38 -06:00 |
|
Jaret Burkett
|
d14f6e567a
|
Allow individual models to scale the loss after it is calculated.
|
2026-06-25 10:13:58 -06:00 |
|
Jaret Burkett
|
7a089fd0d7
|
Add support for training directly on Krea2 Turbo with a training adapter
|
2026-06-23 20:32:02 -06:00 |
|
Jaret Burkett
|
a803611ec1
|
Enable tiling on vae when decoding with low_vram flag on krea2
|
2026-06-23 18:38:52 -06:00 |
|
Jaret Burkett
|
724e67d634
|
Set krea 2 to use new lokr format
|
2026-06-23 13:31:50 -06:00 |
|
Jaret Burkett
|
e20b42e84a
|
Add offloading support for krea2
|
2026-06-23 10:43:36 -06:00 |
|
Jaret Burkett (Ostris)
|
99be3d96a2
|
Add support for Krea2 (#906)
* Add support for krea2
* Update repo pointer to actual repo
|
2026-06-23 09:18:14 -06:00 |
|
Jaret Burkett
|
820d534d6e
|
Add features for models that may need a non masked loss such as inpainiting.
|
2026-06-22 10:59:17 -06:00 |
|
Jaret Burkett
|
99f8fd44e3
|
Add a fallback bbox adjustment when json parsing fails on ideogram4 captioner
|
2026-06-19 07:25:54 -06:00 |
|
Jaret Burkett
|
022d1c29e0
|
Remove color pallet for individual objects in the caption prompts
|
2026-06-19 06:56:18 -06:00 |
|
Jaret Burkett
|
60c1ac6a50
|
Add support for Boogu Image and Boogu Image Edit
|
2026-06-18 15:05:49 -06:00 |
|
Jaret Burkett
|
e886745051
|
Added additional information on addine new models and some additional gotchas
|
2026-06-18 15:04:50 -06:00 |
|
Jaret Burkett
|
515b0ea5cd
|
Remove transformer log supression from captioner
|
2026-06-18 10:00:30 -06:00 |
|
Jaret Burkett
|
ad49d4ef25
|
Update example to cover some common issues
|
2026-06-18 07:54:37 -06:00 |
|
Jaret Burkett
|
66f7c06742
|
Patch away the qwen3vl conv3d that only has slow bf16 kernels
|
2026-06-17 14:29:20 -06:00 |
|
Jaret Burkett
|
92814f9e6d
|
Apply ideogram dynamic shifting to sampling
|
2026-06-16 18:52:16 -06:00 |
|
Jaret Burkett
|
178eb5fbbe
|
Add unconditional lora support so Ideogram 4 inference will more closely resemble the full pipeline results. I pushed a finetuned unconditional lora to the hub as an adapter.
|
2026-06-16 13:27:43 -06:00 |
|
Jaret Burkett
|
f6c0104f25
|
Handle unconditional conditioning for ideogram 4 more in line with example code.
|
2026-06-16 10:23:08 -06:00 |
|
Jaret Burkett
|
86b19589a0
|
Update the Ideogram 4 prompt generation/parsing/ui to handle the updated format notes better.
|
2026-06-16 09:44:38 -06:00 |
|
Jaret Burkett
|
fcccc0fbd2
|
Add gradient checkpointing to ideogram4 vae
|
2026-06-15 10:00:40 -06:00 |
|
Jaret Burkett
|
c0d600b5d6
|
Allow using flash backend for ideogram
|
2026-06-13 15:17:20 -06:00 |
|
Jaret Burkett
|
c8cd78b1a4
|
Allow nested transformer block names for quantization, lora targeting, quantizing
|
2026-06-13 14:33:16 -06:00 |
|
Jaret Burkett
|
6c3b82696e
|
Add support for PRX Pixel T2I
|
2026-06-13 09:47:53 -06:00 |
|
Jaret Burkett
|
a01c83073a
|
Added an example model with docs so people and agents can add models easier.
|
2026-06-13 08:17:06 -06:00 |
|
Jaret Burkett
|
0f580f0663
|
Add compiling option for captioning. Added progress bar for captioning on active job widget. Track steps on caption. Added reverse proxy iframe that plugins can use to add ui functionality.
|
2026-06-12 10:26:52 -06:00 |
|
Jaret Burkett
|
88127557f5
|
Comment out logging supression that was keeping weight loadings from being shown with transformers library
|
2026-06-11 07:53:41 -06:00 |
|
Jaret Burkett
|
cac3815b2c
|
Make DOP run in a single backward pass, should be faster and more stable. Show dop loss on ui
|
2026-06-08 13:55:47 -06:00 |
|
Jaret Burkett
|
35588726de
|
Fixed issue where a buffer was stuck on cpu when offloading ideogram4
|
2026-06-07 13:31:03 -06:00 |
|
Jaret Burkett
|
8968e41234
|
Set ideogram to use new loke saving format
|
2026-06-07 09:49:57 -06:00 |
|
Jaret Burkett
|
c994398850
|
Rework prompts and captioning systems for ideogram to more strictly match the format provided by ideogram.
|
2026-06-05 11:14:36 -06:00 |
|
Jaret Burkett
|
ef12260b80
|
Add a prompt upsample ui for upsampling prompts to ideogram format prompts.
|
2026-06-05 09:56:02 -06:00 |
|
Jaret Burkett
|
edcc7415d1
|
Added an autocaptioner for Ideogram 4 captions
|
2026-06-04 13:04:23 -06:00 |
|
Jaret Burkett
|
6a8d9333b6
|
Improved the prompt handeling of ideogram4 model. Now used advanced prompts class to store them smaller and allow longer prompts
|
2026-06-04 13:00:23 -06:00 |
|
Jaret Burkett
|
63b3181262
|
Added experimental support for Ideogram 4
|
2026-06-04 09:03:00 -06:00 |
|
Jaret Burkett
|
5e84bf0d0b
|
Fixed issue with hidream-01 that could cause a weird nan state. Took forever to track down as it was 1 in 10 starts.
|
2026-05-28 12:48:24 -06:00 |
|
Jaret Burkett
|
ef57c1077c
|
Change z image divisibility
|
2026-05-28 09:09:36 -06:00 |
|
Jaret Burkett
|
68e9b38220
|
Add ability to trigger a save from the ui which whill make the trainer save on the next step
|
2026-05-26 08:38:49 -06:00 |
|
Jaret Burkett
|
7354def271
|
Added decode latent method to qwen image model
|
2026-05-25 10:02:09 -06:00 |
|
Jaret Burkett
|
fbac1cb7f5
|
Dont force flash attention on hidream 01. Causes random issues and is slower.
|
2026-05-24 16:05:22 -06:00 |
|
Jaret Burkett
|
12304e170f
|
Added some experimental loss targets
|
2026-05-24 14:13:23 -06:00 |