599 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Jaret Burkett	c8049a483d	Fix issue with auth token not on build time	2025-03-20 17:27:30 -06:00
Jaret Burkett	f5aa4232fa	Added ability to quantize with torchao	2025-03-20 16:28:54 -06:00
Jaret Burkett	3a6b24f4c8	Added a way to secure the UI. Plus various bug fixes and quality of life updates	2025-03-20 08:07:09 -06:00
Jaret Burkett	bbfd6ef0fe	Fixed bug that prevented using schnell training adapter	2025-03-19 10:25:12 -06:00
Jaret Burkett	b829983b16	Added ability to load video datasets and train with them	2025-03-19 09:54:26 -06:00
Jaret Burkett	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	2025-03-17 13:25:01 -06:00
Jaret Burkett	5eb627dd9d	Add targeted flow guidance training for flow based models	2025-03-17 09:21:23 -06:00
Jaret Burkett	604e76d34d	Fix issue with full finetuning wan	2025-03-17 09:17:40 -06:00
Jaret Burkett	6cde96ae5f	Adding files I forgot to stage to the last commit	2025-03-15 11:59:41 -06:00
Jaret Burkett	1be613ed06	Added advanced mode yaml editor to the ui	2025-03-15 11:58:59 -06:00
Jaret Burkett	c52421aab7	Allow clip image to not have processor on dataloader for raw img	2025-03-15 08:27:54 -06:00
Jaret Burkett	3812957bc9	Added ability to train control loras. Other important bug fixes thrown in	2025-03-14 18:03:00 -06:00
Jaret Burkett	391329dbdc	Fix issue with device placement on te	2025-03-13 20:48:12 -06:00
Jaret Burkett	3b45892b4f	Update sponsors	2025-03-13 19:06:59 -06:00
Jaret Burkett	cf4216e6b8	Add support for wan training in ui	2025-03-13 18:54:27 -06:00
Jaret Burkett	31e057d9a3	Fixed issue with device placement in some scenereos when doing low vram on wan	2025-03-13 10:30:27 -06:00
Jaret Burkett	d507b44a7b	Update sponsors	2025-03-08 22:28:20 -07:00
Jaret Burkett	242c04a0b8	Fix error with training video models with batch greater than 1	2025-03-08 18:47:27 -07:00
Jaret Burkett	386e68a422	Fixed a bug that changes all samples to webp	2025-03-08 18:02:56 -07:00
Jaret Burkett	850b8da6e5	Added siglip 2 vision encoder for custom adapter	2025-03-09 00:14:44 +00:00
Jaret Burkett	51ad19b568	Add config file examples for training Wan LoRAs on 24GB cards.	2025-03-08 13:56:21 -07:00
Jaret Burkett	e6739f7eb2	Convert wan lora weights on save to be something comfy can handle	2025-03-08 12:55:11 -07:00
Jaret Burkett	7e37918fbc	Double tap module casting as it doesent seem to happen every time.	2025-03-07 22:15:24 -07:00
Jaret Burkett	4d88f8f218	Fixed cuda error when not all tensors have been moved to the correct device.	2025-03-07 22:04:35 -07:00
Jaret Burkett	25341c4613	Got wan 14b training to work on 24GB card.	2025-03-07 17:04:10 -07:00
Jaret Burkett	391cf80fea	Added training for Wan2.1. Not finalized, wait.	2025-03-07 13:53:44 -07:00
Jaret Burkett	4e3bda7c70	Merge pull request #264 from ostris/cogview4 ... Added basics for CogView4. Broken as hell though. Dont use.	2025-03-05 14:52:06 -07:00
Jaret Burkett	763128ea42	Note about cogview	2025-03-05 14:46:11 -07:00
Jaret Burkett	4fe33f51c1	Fix issue with picking layers for quantization, adjust layers fo better quantization of cogview4	2025-03-05 13:44:40 -07:00
Jaret Burkett	aa44828c0c	WIP more work on cogview4	2025-03-05 09:43:00 -07:00
Jaret Burkett	6f6fb90812	Added cogview4. Loss still needs work.	2025-03-04 18:43:52 -07:00
Jaret Burkett	c57434ad7b	Removed wan submodule stuff for now	2025-03-04 00:32:24 -07:00
Jaret Burkett	8bb47d1bfe	Merge branch 'main' into wan21	2025-03-04 00:31:57 -07:00
Jaret Burkett	e7dbb20f68	Removed wan submodule for now	2025-03-04 00:29:19 -07:00
Jaret Burkett	c5e0c2bbe2	Fixes to allow for redux assisted training	2025-03-03 16:27:19 -07:00
Jaret Burkett	1f3f45a48d	Bugfixes	2025-03-03 08:22:15 -07:00
Jaret Burkett	3c8c84f156	Added supporters to readme and a script to update it	2025-03-02 10:25:27 -07:00
Jaret Burkett	b001d77efb	Added LoKr instructions to the readme	2025-03-02 08:55:56 -07:00
Jaret Burkett	7ae31c9ae9	Added LoKr to the ui	2025-03-02 08:49:01 -07:00
Jaret Burkett	b16819f8e7	Added LoKr support	2025-03-02 06:57:50 -07:00
Jaret Burkett	f5e40dfa62	WIP on wan	2025-03-01 16:12:52 -07:00
Jaret Burkett	acc79956aa	WIP create new class to add new models more easily	2025-03-01 13:49:02 -07:00
Jaret Burkett	60539c0b0f	Allow using prior loss with a custom adapter	2025-03-01 08:01:14 -07:00
Jaret Burkett	dd700f70b3	Avoid loading state dict for automagic for now until I can sort out some issues	2025-02-26 17:03:14 -07:00
Jaret Burkett	d360e76661	fixed issue with dop prompt replacement	2025-02-26 13:35:18 -07:00
Jaret Burkett	6ec23ed226	Fixed issue when doing inverted masked prior with flowmatching algos	2025-02-26 12:12:32 -07:00
Jaret Burkett	f6e16e582a	Added Differential Output Preservation Loss to trainer and ui	2025-02-25 20:12:36 -07:00
Jaret Burkett	259ded9602	Fixed issue with trigger word saving in ui	2025-02-24 11:04:24 -07:00
Jaret Burkett	440ba5fb3d	Spawn windows in an cmd terminal. Should be working now, but not sure on my system	2025-02-24 08:54:56 -07:00
Jaret Burkett	093f14ac19	UI Bug fixes and initial windows support	2025-02-24 08:15:22 -07:00