612 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Jaret Burkett	6021a3dbc0	Change inpainting mask to zero out on latents instead of image for inpaint area.	2025-03-24 14:16:52 -06:00
Jaret Burkett	71d7a52146	Fixed issue with python being wrong on docker	2025-03-24 14:16:09 -06:00
Jaret Burkett	45be82d5d6	Handle inpainting training for control_lora adapter	2025-03-24 13:17:47 -06:00
Jaret Burkett	f10937e6da	Handle multi control inputs for control lora training	2025-03-23 07:37:08 -06:00
Jaret Burkett	ccb66c748f	Update readme install directions	2025-03-22 15:14:04 -06:00
Jaret Burkett	2aca2883e7	Update windows install directions for new version of torch	2025-03-22 15:12:51 -06:00
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