606 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Jaret Burkett	1ad58c5816	Changed control lora to only have new weights and leave other input weights alone for more flexability of using multiple ones together.	2025-03-22 10:24:52 -06:00
Jaret Burkett	6dea41b9fc	Version bump	2025-03-21 11:48:03 -06:00
Jaret Burkett	9a902c067f	Show the console log on the job overview in the ui.	2025-03-21 11:45:36 -06:00
Jaret Burkett	0bbc69c135	Fix issue where the job would hang in the ui if it failed to start	2025-03-21 09:46:41 -06:00
Jaret Burkett	6c5eb0cf87	Move pytorch install above cache bust to prevent reinstalling and reuploading it	2025-03-21 06:43:09 -06:00
Jaret Burkett	e3373671b9	Fixed missing dependency	2025-03-21 06:25:30 -06:00
Jaret Burkett	aceb3a0f25	Rework docker	2025-03-20 20:07:27 -06:00
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