Commit Graph

Select branches

Hide Pull Requests

WIP

dev

development

hidream_o1

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

#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

#564

#567

#578

#580

#581

#583

#594

#599

#60

#600

#601

#601

#606

#608

#61

#610

#611

#614

#616

#629

#630

#633

#637

#639

#644

#646

#648

#649

#655

#658

#660

#661

#666

#666

#668

#668

#673

#682

#685

#687

#69

#692

#694

#694

#695

#697

#698

#699

#7

#702

#703

#705

#706

#707

#708

#709

#709

#713

#713

#717

#718

#718

#719

#720

#721

#724

#724

#726

#727

#727

#732

#732

#733

#738

#742

#743

#743

#745

#748

#749

#754

#759

#760

#761

#762

#763

#765

#765

#766

#766

#767

#768

#769

#770

#774

#774

#777

#777

#778

#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

#828

#829

#829

#86

#9

#95

Mono Color

• 6c5eb0cf87 Move pytorch install above cache bust to prevent reinstalling and reuploading it Jaret Burkett 2025-03-21 06:43:09 -06:00
• e3373671b9 Fixed missing dependency Jaret Burkett 2025-03-21 06:25:30 -06:00
• aceb3a0f25 Rework docker Jaret Burkett 2025-03-20 20:07:27 -06:00
• c8049a483d Fix issue with auth token not on build time Jaret Burkett 2025-03-20 17:27:30 -06:00
• f5aa4232fa Added ability to quantize with torchao Jaret Burkett 2025-03-20 16:28:54 -06:00
• 3a6b24f4c8 Added a way to secure the UI. Plus various bug fixes and quality of life updates Jaret Burkett 2025-03-20 08:07:09 -06:00
• bbfd6ef0fe Fixed bug that prevented using schnell training adapter Jaret Burkett 2025-03-19 10:25:12 -06:00
• b829983b16 Added ability to load video datasets and train with them Jaret Burkett 2025-03-19 09:54:26 -06:00
• fa187b1208 Added differential masking to targeted_flow_guidance to allow the model learn to clean up the targeted area a little more than unmasked was capable of Jaret Burkett 2025-03-17 13:25:01 -06:00
• 5eb627dd9d Add targeted flow guidance training for flow based models Jaret Burkett 2025-03-17 09:21:23 -06:00
• 604e76d34d Fix issue with full finetuning wan Jaret Burkett 2025-03-17 09:17:40 -06:00
• 6cde96ae5f Adding files I forgot to stage to the last commit Jaret Burkett 2025-03-15 11:59:41 -06:00
• 1be613ed06 Added advanced mode yaml editor to the ui Jaret Burkett 2025-03-15 11:58:59 -06:00
• c52421aab7 Allow clip image to not have processor on dataloader for raw img Jaret Burkett 2025-03-15 08:27:54 -06:00
• 3812957bc9 Added ability to train control loras. Other important bug fixes thrown in Jaret Burkett 2025-03-14 18:03:00 -06:00
• 391329dbdc Fix issue with device placement on te Jaret Burkett 2025-03-13 20:48:12 -06:00
• 3b45892b4f Update sponsors Jaret Burkett 2025-03-13 19:06:59 -06:00
• cf4216e6b8 Add support for wan training in ui Jaret Burkett 2025-03-13 18:54:27 -06:00
• 31e057d9a3 Fixed issue with device placement in some scenereos when doing low vram on wan Jaret Burkett 2025-03-13 10:30:27 -06:00
• d507b44a7b Update sponsors Jaret Burkett 2025-03-08 22:28:20 -07:00
• 242c04a0b8 Fix error with training video models with batch greater than 1 Jaret Burkett 2025-03-08 18:47:27 -07:00
• 386e68a422 Fixed a bug that changes all samples to webp Jaret Burkett 2025-03-08 18:02:56 -07:00
• 850b8da6e5 Added siglip 2 vision encoder for custom adapter Jaret Burkett 2025-03-09 00:14:44 +00:00
• 51ad19b568 Add config file examples for training Wan LoRAs on 24GB cards. Jaret Burkett 2025-03-08 13:56:21 -07:00
• e6739f7eb2 Convert wan lora weights on save to be something comfy can handle Jaret Burkett 2025-03-08 12:55:11 -07:00
• 7e37918fbc Double tap module casting as it doesent seem to happen every time. Jaret Burkett 2025-03-07 22:15:24 -07:00
• 4d88f8f218 Fixed cuda error when not all tensors have been moved to the correct device. Jaret Burkett 2025-03-07 22:04:35 -07:00
• 25341c4613 Got wan 14b training to work on 24GB card. Jaret Burkett 2025-03-07 17:04:10 -07:00
• 391cf80fea Added training for Wan2.1. Not finalized, wait. Jaret Burkett 2025-03-07 13:53:44 -07:00
• 4e3bda7c70 Merge pull request #264 from ostris/cogview4 Jaret Burkett 2025-03-05 14:52:06 -07:00
• 763128ea42 Note about cogview Jaret Burkett 2025-03-05 14:46:11 -07:00
• 4fe33f51c1 Fix issue with picking layers for quantization, adjust layers fo better quantization of cogview4 Jaret Burkett 2025-03-05 13:44:40 -07:00
• aa44828c0c WIP more work on cogview4 Jaret Burkett 2025-03-05 09:43:00 -07:00
• 6f6fb90812 Added cogview4. Loss still needs work. Jaret Burkett 2025-03-04 18:43:52 -07:00
• c57434ad7b Removed wan submodule stuff for now Jaret Burkett 2025-03-04 00:32:24 -07:00
• 8bb47d1bfe Merge branch 'main' into wan21 Jaret Burkett 2025-03-04 00:31:57 -07:00
• e7dbb20f68 Removed wan submodule for now Jaret Burkett 2025-03-04 00:29:19 -07:00
• c5e0c2bbe2 Fixes to allow for redux assisted training Jaret Burkett 2025-03-03 16:27:19 -07:00
• 1f3f45a48d Bugfixes Jaret Burkett 2025-03-03 08:22:15 -07:00
• 3c8c84f156 Added supporters to readme and a script to update it Jaret Burkett 2025-03-02 10:25:27 -07:00
• b001d77efb Added LoKr instructions to the readme Jaret Burkett 2025-03-02 08:55:56 -07:00
• 7ae31c9ae9 Added LoKr to the ui Jaret Burkett 2025-03-02 08:49:01 -07:00
• b16819f8e7 Added LoKr support Jaret Burkett 2025-03-02 06:57:50 -07:00
• f5e40dfa62 WIP on wan Jaret Burkett 2025-03-01 16:12:52 -07:00
• acc79956aa WIP create new class to add new models more easily Jaret Burkett 2025-03-01 13:49:02 -07:00
• 60539c0b0f Allow using prior loss with a custom adapter Jaret Burkett 2025-03-01 08:01:14 -07:00
• dd700f70b3 Avoid loading state dict for automagic for now until I can sort out some issues Jaret Burkett 2025-02-26 17:03:14 -07:00
• d360e76661 fixed issue with dop prompt replacement Jaret Burkett 2025-02-26 13:35:18 -07:00
• 6ec23ed226 Fixed issue when doing inverted masked prior with flowmatching algos Jaret Burkett 2025-02-26 12:12:32 -07:00
• f6e16e582a Added Differential Output Preservation Loss to trainer and ui Jaret Burkett 2025-02-25 20:12:36 -07:00
• 259ded9602 Fixed issue with trigger word saving in ui Jaret Burkett 2025-02-24 11:04:24 -07:00
• 440ba5fb3d Spawn windows in an cmd terminal. Should be working now, but not sure on my system Jaret Burkett 2025-02-24 08:54:56 -07:00
• 093f14ac19 UI Bug fixes and initial windows support Jaret Burkett 2025-02-24 08:15:22 -07:00
• f0fbd8bb53 Merge pull request #256 from ostris/ui Jaret Burkett 2025-02-23 16:10:49 -07:00
• 0a981bea2b Fixed typo Jaret Burkett 2025-02-23 16:07:38 -07:00
• 1d0e3a4498 Fixed some build issues for now. Added info to the readme Jaret Burkett 2025-02-23 15:59:17 -07:00
• 3c7daf49f3 Add HF token to env when spawing via ui Jaret Burkett 2025-02-23 14:52:12 -07:00
• 56d8d6bd81 Capture speed from the timer for the ui Jaret Burkett 2025-02-23 14:38:46 -07:00
• 3e49337a58 Set step to the last step saved at when exiting Jaret Burkett 2025-02-23 13:21:22 -07:00
• 60f848a877 Send more data when loading the model to the ui Jaret Burkett 2025-02-23 12:49:54 -07:00
• b366e46f1c Added more settings to the training config Jaret Burkett 2025-02-23 12:34:52 -07:00
• a280f78c69 Added checkpoint downloader Jaret Burkett 2025-02-22 16:48:15 -07:00
• 6e19e7449e Fixed some issues with gpu info refreshing Jaret Burkett 2025-02-22 14:14:23 -07:00
• a6d46ad9ae Cleanup of job page Jaret Burkett 2025-02-22 13:54:06 -07:00
• f3725578dd Cleaned up dashboard Jaret Burkett 2025-02-22 13:23:26 -07:00
• ed99c3c0c8 Moved gpu to its own widget Jaret Burkett 2025-02-22 12:43:20 -07:00
• ed84c19205 Moved the job action bar to a shred component Jaret Burkett 2025-02-22 12:20:14 -07:00
• a7a9c11d9e Fixed add image dropbox Jaret Burkett 2025-02-22 11:59:21 -07:00
• f60698d0ee Fixed some bugs with ui and lock job name to prevent issues with continuing training. Jaret Burkett 2025-02-22 11:49:36 -07:00
• 5f094fb17a Added controls to the jobs table Jaret Burkett 2025-02-22 10:57:53 -07:00
• a5227cba7b Switched to a universal table library Jaret Burkett 2025-02-22 09:59:17 -07:00
• 77a5e01301 Added proper icon Jaret Burkett 2025-02-22 09:05:55 -07:00
• 4ef5a668c0 Make left arrow browsing only hit last image max Jaret Burkett 2025-02-21 22:04:53 -07:00
• f081d14527 Preview samples full screen and use arrow keys to navigate them Jaret Burkett 2025-02-21 21:52:24 -07:00
• 710c6de1c9 Samples work in ui now Jaret Burkett 2025-02-21 20:28:52 -07:00
• 2b6e66e0cb Mor ui work Jaret Burkett 2025-02-21 12:40:17 -07:00
• ab641e014f Added funding github stuff Jaret Burkett 2025-02-21 17:13:36 +00:00
• ad87f72384 Start, stop, monitor jobs from ui working. Jaret Burkett 2025-02-21 09:49:28 -07:00
• d0214c0df9 Make ui more uniform Jaret Burkett 2025-02-21 06:18:27 -07:00
• adcf884c0f Built out the ui trainer plugin with db comminication Jaret Burkett 2025-02-21 05:53:35 -07:00
• f778d979b5 Saving captions is working Jaret Burkett 2025-02-20 16:17:00 -07:00
• db3ccbba33 Handle image deletion Jaret Burkett 2025-02-20 15:58:10 -07:00
• 0d2be18a9b Delete datasets Jaret Burkett 2025-02-20 14:49:03 -07:00
• bbc340e545 Cleanup and add hooks Jaret Burkett 2025-02-20 13:38:58 -07:00
• 33fdfd6091 Added beginning or lokr Jaret Burkett 2025-02-20 12:47:42 -07:00
• 9f6030620f Dataset uploads working Jaret Burkett 2025-02-20 12:47:01 -07:00
• b5252b5028 More ui work Jaret Burkett 2025-02-20 11:19:01 -07:00
• b0d8fc220d More ui more ui Jaret Burkett 2025-02-19 20:54:02 -07:00
• cef7d9e594 Config ui section is coming along Jaret Burkett 2025-02-19 07:52:24 -07:00
• b13fcc1039 Setup a very basic ui Jaret Burkett 2025-02-18 10:57:14 -07:00
• b32d7e552b Shamelessly beg for money Jaret Burkett 2025-02-18 05:15:29 -07:00
• 4af6c5cf30 Work on supporting flex.2 potential arch Jaret Burkett 2025-02-17 14:10:25 -07:00
• 1f7784510d WIP Flex 2 pipeline Jaret Burkett 2025-02-16 14:54:29 -07:00
• 87e557cf1e Bug fixes and improvements to llmadapter Jaret Burkett 2025-02-15 07:18:07 -07:00
• bd8d7dc081 fixed various issues with llm attention masking. Added block training on the llm adapter. Jaret Burkett 2025-02-14 11:24:01 -07:00
• 2be6926398 Added back syustem prompt for llm and remove those tokens from the embeddings Jaret Burkett 2025-02-14 07:23:37 -07:00
• 87ac031859 Remove system prompt, shouldnt be necessary fo rhow it works. Jaret Burkett 2025-02-13 08:42:48 -07:00
• 7679105d52 Added llm text encoder adapter Jaret Burkett 2025-02-13 08:28:32 -07:00
• 2622de1e01 DFE tweaks. Adding support for more llms as text encoders Jaret Burkett 2025-02-13 04:31:49 -07:00
• 8450aca10e Fixed missed merge conflice and locked diffusers version Jaret Burkett 2025-02-12 09:40:02 -07:00