Commit Graph

Select branches

Hide Pull Requests

WIP

dev

development

kohya-sdxl

lumina2

main

multi-gpu

qwen_image_edit

sdxl

wavelet_loss

#107

#109

#115

#12

#120

#124

#125

#128

#138

#141

#143

#155

#156

#158

#160

#160

#168

#173

#176

#177

#179

#183

#184

#185

#191

#196

#224

#226

#232

#233

#244

#245

#246

#249

#25

#251

#253

#253

#256

#263

#264

#265

#271

#273

#274

#275

#278

#283

#290

#300

#302

#302

#303

#304

#307

#318

#326

#326

#331

#333

#341

#342

#343

#350

#353

#354

#359

#362

#373

#375

#377

#381

#382

#383

#385

#392

#392

#395

#407

#41

#417

#418

#420

#422

#426

#427

#430

#431

#432

#434

#442

#446

#447

#452

#455

#459

#461

#464

#465

#467

#469

#473

#474

#488

#489

#492

#494

#496

#498

#499

#500

#509

#511

#513

#519

#520

#521

#522

#523

#525

#527

#540

#547

#551

#555

#558

#559

#563

#563

#564

#567

#578

#580

#581

#583

#594

#599

#60

#600

#601

#606

#608

#61

#610

#611

#614

#616

#629

#630

#633

#637

#639

#644

#646

#648

#649

#655

#658

#660

#661

#666

#668

#673

#682

#685

#687

#69

#692

#694

#694

#695

#697

#698

#699

#7

#702

#703

#705

#706

#707

#708

#709

#713

#717

#718

#718

#719

#720

#721

#724

#726

#727

#727

#732

#732

#733

#738

#742

#743

#745

#748

#749

#754

#759

#760

#761

#762

#763

#765

#765

#766

#766

#767

#768

#769

#770

#774

#774

#777

#777

#778

#782

#782

#783

#785

#793

#80

#800

#803

#803

#805

#805

#807

#807

#808

#808

#809

#811

#811

#812

#812

#813

#813

#814

#814

#819

#819

#825

#825

#827

#828

#829

#829

#830

#830

#831

#833

#837

#837

#838

#838

#840

#840

#841

#841

#842

#842

#844

#845

#846

#846

#849

#850

#850

#851

#851

#856

#856

#86

#860

#860

#863

#863

#864

#864

#866

#867

#868

#873

#876

#876

#877

#878

#880

#881

#882

#884

#884

#885

#885

#886

#886

#887

#888

#888

#890

#890

#891

#892

#893

#894

#895

#896

#896

#897

#898

#899

#9

#900

#900

#901

#901

#902

#902

#903

#904

#906

#909

#910

#910

#911

#911

#912

#912

#913

#913

#914

#915

#915

#916

#916

#917

#917

#919

#919

#95

Mono Color

• 4e50535478 Allow models to provide an additional loss main Jaret Burkett 2026-06-28 09:49:36 -06:00
• 1e12b6b73f Remove duplicate save when triggering save next step Jaret Burkett 2026-06-28 07:18:16 -06:00
• 7ee1f98f6d Allow loading and saving z_image in comfy safetensors format. Jaret Burkett 2026-06-28 07:15:49 -06:00
• c97fc9973a Fix loading pretraied lora when merging network on save Jaret Burkett 2026-06-28 06:04:36 -06:00
• f8667f0334 Save Krea2 even if quantized Jaret Burkett 2026-06-28 06:03:37 -06:00
• df6ea4263d Version bump Jaret Burkett 2026-06-26 13:02:10 -06:00
• ad87aacec0 Add ability to set certain layers to full for loras Jaret Burkett 2026-06-26 13:01:48 -06:00
• 4a99ddabad Fix breaking change with diffusers qwen image Jaret Burkett 2026-06-26 10:04:35 -06:00
• 5f04ae7ad5 Rework merge_network_on_save to handle dequantization on merging and saving for much more efficient full finetuning. Jaret Burkett 2026-06-25 13:19:38 -06:00
• 6ecff36f26 Add a way to do full rank lora modules of non lora layers such as embeddings, norm, etc Jaret Burkett 2026-06-25 11:55:07 -06:00
• 4eb0707639 Add a control generation script. Jaret Burkett 2026-06-25 10:14:24 -06:00
• d14f6e567a Allow individual models to scale the loss after it is calculated. Jaret Burkett 2026-06-25 10:13:58 -06:00
• f743ccf7ef Save before last sample Jaret Burkett 2026-06-24 09:13:01 -06:00
• 089e41dd1c Compile improvements - auto cache size, fix fullgraph setting, fix triton detection (#899) PlagueKind 2026-06-24 15:59:41 +02:00
• d586125b40 Force HF downloads to show progress bars when output is not a TTY (#909) fatalis 2026-06-24 08:55:38 -05:00
• 7a089fd0d7 Add support for training directly on Krea2 Turbo with a training adapter Jaret Burkett 2026-06-23 20:32:02 -06:00
• a803611ec1 Enable tiling on vae when decoding with low_vram flag on krea2 Jaret Burkett 2026-06-23 18:38:52 -06:00
• 724e67d634 Set krea 2 to use new lokr format Jaret Burkett 2026-06-23 13:31:50 -06:00
• e20b42e84a Add offloading support for krea2 Jaret Burkett 2026-06-23 10:43:36 -06:00
• 99be3d96a2 Add support for Krea2 (#906) Jaret Burkett (Ostris) 2026-06-23 09:18:14 -06:00
• af594061ab Add ability to do hidden states with tipsv2 Jaret Burkett 2026-06-22 11:00:04 -06:00
• 820d534d6e Add features for models that may need a non masked loss such as inpainiting. Jaret Burkett 2026-06-22 10:59:17 -06:00
• c133c55cf5 Fix issue with mask generator revision Jaret Burkett 2026-06-22 08:19:27 -06:00
• d51463ca52 Add ability to recover from a truncated image file in the dataset Jaret Burkett 2026-06-22 05:07:30 -06:00
• ba0b3dbb65 Force batch size when bucket is too small by duplicating items in the batch Jaret Burkett 2026-06-21 20:01:29 -06:00
• dba092fc15 Keep a real git repo in the docker image Jaret Burkett 2026-06-21 06:54:19 -06:00
• 548a286992 Add captioning monitoring information to the dataset page so you can just stay on that page with all the info there. Jaret Burkett 2026-06-19 08:00:40 -06:00
• 99f8fd44e3 Add a fallback bbox adjustment when json parsing fails on ideogram4 captioner Jaret Burkett 2026-06-19 07:25:54 -06:00
• 4af4fb9d58 Add auto_frame_count support for all remaining video models (#897) fatalis 2026-06-19 08:05:18 -05:00
• 022d1c29e0 Remove color pallet for individual objects in the caption prompts Jaret Burkett 2026-06-19 06:56:18 -06:00
• 60c1ac6a50 Add support for Boogu Image and Boogu Image Edit Jaret Burkett 2026-06-18 15:05:49 -06:00
• e886745051 Added additional information on addine new models and some additional gotchas Jaret Burkett 2026-06-18 15:04:50 -06:00
• 515b0ea5cd Remove transformer log supression from captioner Jaret Burkett 2026-06-18 10:00:30 -06:00
• e8c828089a Fix issue where gpu sometimes doesnt show on caption modal Jaret Burkett 2026-06-18 09:37:59 -06:00
• ad49d4ef25 Update example to cover some common issues Jaret Burkett 2026-06-18 07:54:37 -06:00
• 66f7c06742 Patch away the qwen3vl conv3d that only has slow bf16 kernels Jaret Burkett 2026-06-17 14:29:20 -06:00
• 92814f9e6d Apply ideogram dynamic shifting to sampling Jaret Burkett 2026-06-16 18:52:16 -06:00
• 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. Jaret Burkett 2026-06-16 13:27:43 -06:00
• f6c0104f25 Handle unconditional conditioning for ideogram 4 more in line with example code. Jaret Burkett 2026-06-16 10:23:08 -06:00
• 86b19589a0 Update the Ideogram 4 prompt generation/parsing/ui to handle the updated format notes better. Jaret Burkett 2026-06-16 09:44:38 -06:00
• fcccc0fbd2 Add gradient checkpointing to ideogram4 vae Jaret Burkett 2026-06-15 10:00:40 -06:00
• c730d64478 Added a flag to keep loading the image when latents are cached. Useful for DFE and other methods that target pixelspace losses. Jaret Burkett 2026-06-15 05:31:48 -06:00
• faa770fc79 Another docker thing? Jaret Burkett 2026-06-14 12:27:05 -06:00
• 570c806924 Mode docker build work Jaret Burkett 2026-06-14 12:24:55 -06:00
• ebbb09230b Add requirements base to docker build Jaret Burkett 2026-06-14 12:21:28 -06:00
• 5df3fb69e3 Rework Docker image for a minimal build/pull/push footprint Jaret Burkett 2026-06-14 12:19:58 -06:00
• c0d600b5d6 Allow using flash backend for ideogram Jaret Burkett 2026-06-13 15:17:20 -06:00
• c8cd78b1a4 Allow nested transformer block names for quantization, lora targeting, quantizing Jaret Burkett 2026-06-13 14:33:16 -06:00
• 17c9279828 Version bump Jaret Burkett 2026-06-13 09:48:26 -06:00
• 6c3b82696e Add support for PRX Pixel T2I Jaret Burkett 2026-06-13 09:47:53 -06:00
• a01c83073a Added an example model with docs so people and agents can add models easier. Jaret Burkett 2026-06-13 08:17:06 -06:00
• 2f91db8363 Defauly to compiling full graph to false Jaret Burkett 2026-06-13 07:28:51 -06:00
• e908d85f5e Allow quantized unet offload compile and force fullgraph false (#881) PlagueKind 2026-06-13 15:27:28 +02:00
• 0165fb2ac6 Update npm packages Jaret Burkett 2026-06-12 17:50:47 -06:00
• c90c400716 Add Ustris Cloud info to the README Jaret Burkett 2026-06-12 12:53:47 -06:00
• 43b22b91ee Add a triton check on compile Jaret Burkett 2026-06-12 12:26:16 -06:00
• 10e50d5797 Fix issue with casting unet after compilation Jaret Burkett 2026-06-12 12:16:51 -06:00
• d83f7dd4d9 Fix a few issues with compile. Changed defaults. Future proofed block layer compile. Jaret Burkett 2026-06-12 11:43:43 -06:00
• a5558ae7d9 Add compile to the actual right config section. Jaret Burkett 2026-06-12 11:12:41 -06:00
• c09b228a35 Add model compiling to the ui Jaret Burkett 2026-06-12 11:02:10 -06:00
• 6b1f89f30b Enhanced torch.compile System with Block-Level Compilation and Unified Whole-Model Fallback (#866) PlagueKind 2026-06-12 18:35:12 +02:00
• 324faf17b3 accept .json uploads (#880) Rainer 2026-06-12 19:31:19 +03:00
• 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. Jaret Burkett 2026-06-12 10:26:52 -06:00
• 3fd14f3805 Change suporters view to an auto updating SVG file so the README doesnt need to be updated constantly. Jaret Burkett 2026-06-12 09:48:50 -06:00
• 9cf34f945c Allow multi selecting jobs and deleting them in one go Jaret Burkett 2026-06-12 08:01:56 -06:00
• 55ce6570f2 Automagic 3 rework. Stable in my testing. Jaret Burkett 2026-06-12 07:52:44 -06:00
• 53ebb93edb fix learning rate metric truncating to zero on graph (#877) fatalis 2026-06-12 08:30:21 -05:00
• 88127557f5 Comment out logging supression that was keeping weight loadings from being shown with transformers library Jaret Burkett 2026-06-11 07:53:41 -06:00
• 01b6a9806b Another complete rework of automagic3. Added a decay to the LR spread to the mean to prevent LRs fighting with eachother Jaret Burkett 2026-06-09 12:08:45 -06:00
• 9e99d3ce5d Save loss graph view settings per page to local storage. Jaret Burkett 2026-06-09 07:29:51 -06:00
• acb1548722 Updated the comments and doc for Automagic v3 Jaret Burkett 2026-06-09 07:05:48 -06:00
• a1ac6e8b01 Reworked automagic v3 again. Seems more stable. Still testing. Jaret Burkett 2026-06-08 22:03:54 -06:00
• 5d6887fd98 Major updates to automagic3 optimizer. Seems to be functioning more ideally and naturally decays, as it should. Jaret Burkett 2026-06-08 14:09:02 -06:00
• 0d018db689 Rework the toggles so we can hide the trend line from the loss graph. No need for a raw toggle either. Jaret Burkett 2026-06-08 13:56:37 -06:00
• cac3815b2c Make DOP run in a single backward pass, should be faster and more stable. Show dop loss on ui Jaret Burkett 2026-06-08 13:55:47 -06:00
• 687def6f7a Add better smoothing with ema rounding of the ends of the loss graph for first and last instance do not dominate the smoothing factors. Jaret Burkett 2026-06-08 10:43:46 -06:00
• d7f8887bbf Show all logged metrics on the loss graph Jaret Burkett 2026-06-08 09:43:03 -06:00
• e281df70dd Allow automagic3 to run unfused. Add some clipping. Jaret Burkett 2026-06-08 08:59:06 -06:00
• c9cdbb5bb7 Version bump Jaret Burkett 2026-06-07 16:08:11 -06:00
• c78b1404e3 Deepen offload prefetch pipeline with per-slot events Jaret Burkett 2026-06-07 16:07:13 -06:00
• cdff6e36aa Pin inner stores of torachao to speed up layer offloading for quantized models around 25% Jaret Burkett 2026-06-07 15:52:01 -06:00
• 75781fb5a5 Fix float8 weights not offloading to CPU in layer offloading Jaret Burkett 2026-06-07 15:36:17 -06:00
• 7c1a76f336 Fix text encoder offload bug when caching embeddings (#868) PlagueKind 2026-06-07 22:09:32 +02:00
• 35588726de Fixed issue where a buffer was stuck on cpu when offloading ideogram4 Jaret Burkett 2026-06-07 13:31:03 -06:00
• 1dc9a797cf Added Automagic v3 Jaret Burkett 2026-06-07 12:06:42 -06:00
• 82190b41e6 Fix bug where EMA was not being initialized even when setup in the config. EMA will now properly be setup and used. Jaret Burkett 2026-06-07 11:11:49 -06:00
• 8968e41234 Set ideogram to use new loke saving format Jaret Burkett 2026-06-07 09:49:57 -06:00
• fa0dca288d Version bump Jaret Burkett 2026-06-06 10:24:31 -06:00
• 41157b460c Added ability to set the caption extention in dataset viewer, captioner, and trainer so one dataset can have multiple caption styles in different files with different extensions. Added dataset caption template for a blank ideogram 4 formatted template. Jaret Burkett 2026-06-06 08:32:24 -06:00
• 10cdeb394e Version bump Jaret Burkett 2026-06-05 14:09:46 -06:00
• 21a6beb194 Allow editing the boxes visually on the sample section for ideogram Jaret Burkett 2026-06-05 14:09:18 -06:00
• 4441080c05 Rework the ui of ideogram dataset image caption editor Jaret Burkett 2026-06-05 13:20:58 -06:00
• b70083a74f Fix issue with default sampels. Jaret Burkett 2026-06-05 12:33:42 -06:00
• c994398850 Rework prompts and captioning systems for ideogram to more strictly match the format provided by ideogram. Jaret Burkett 2026-06-05 11:14:36 -06:00
• 6fd2253932 Imporved editing boxed on the dataset viewer Jaret Burkett 2026-06-05 10:30:06 -06:00
• ef12260b80 Add a prompt upsample ui for upsampling prompts to ideogram format prompts. Jaret Burkett 2026-06-05 09:56:02 -06:00
• 90a2084f70 Allow adjusting, adding, and deleting bounding boxes. Jaret Burkett 2026-06-04 20:29:31 -06:00
• bb60f6d1d1 Show bounding boxes on the sample image card if we have them in the prompt Jaret Burkett 2026-06-04 15:59:02 -06:00
• edcc7415d1 Added an autocaptioner for Ideogram 4 captions Jaret Burkett 2026-06-04 13:04:23 -06:00
• 6a8d9333b6 Improved the prompt handeling of ideogram4 model. Now used advanced prompts class to store them smaller and allow longer prompts Jaret Burkett 2026-06-04 13:00:23 -06:00