653 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Jaret Burkett	89c0f688db	Merge branch 'main' into hidream	2025-04-13 21:16:07 -06:00
Jaret Burkett	1e0bff653c	Fix new bug I accidently introduced with lora	2025-04-13 21:15:07 -06:00
Jaret Burkett	3a5ea2c742	Remove some moe stuff for finetuning. Drastically reduces vram usage	2025-04-14 00:57:34 +00:00
Jaret Burkett	f80cf99f40	Hidream is training, but has a memory leak	2025-04-13 23:28:18 +00:00
Jaret Burkett	594e166ca3	Initial support for hidream. Still a WIP	2025-04-13 13:50:11 -06:00
Jaret Burkett	ca3ce0f34c	Make it easier to designate lora blocks for new models. Improve i2v adapter speed. Fix issue with i2v adapter where cached torch tensor was wrong range.	2025-04-13 13:49:13 -06:00
Jaret Burkett	6fb44db6a0	Finished up first frame for i2v adapter	2025-04-12 17:13:04 -06:00
Jaret Burkett	cd37ccfc2e	Use gradient checkpointing on DFE models if set	2025-04-11 10:45:39 -06:00
Jaret Burkett	4a43589666	Use a shuffled embedding as unconditional for i2v adapter	2025-04-11 10:44:43 -06:00
Jaret Burkett	059155174a	Added mask diffirential mask dialation for flex2. Handle video for the i2v adapter	2025-04-10 11:50:01 -06:00
Jaret Burkett	9794416a5d	Fixed bug when loading video datasets	2025-04-10 08:16:05 -06:00
Jaret Burkett	d8bdc03256	Allow full control of caption extensions	2025-04-10 07:42:04 -06:00
Jaret Burkett	96ba2fd129	Added methods to the dataloader to automatically generate controls for line, mask, inpainting, depth, and pose.	2025-04-09 13:35:04 -06:00
Jaret Burkett	615b0d0e94	Added initial support for training i2v adapter WIP	2025-04-09 08:06:29 -06:00
Jaret Burkett	a8680c75eb	Added initial support for finetuning wan i2v WIP	2025-04-07 20:34:38 -06:00
Jaret Burkett	38ad5a4644	Fixed issue with video dataset sizing	2025-04-07 12:46:41 -06:00
Jaret Burkett	6c8b5ab606	Added some more useful error handeling and logging	2025-04-07 08:01:37 -06:00
Jaret Burkett	7c21eac1b3	Added support for Lodestone Rock's Chroma model	2025-04-05 13:21:36 -06:00
Jaret Burkett	2b901cca39	Small tweaks and bug fixes and future proofing	2025-04-05 12:39:45 -06:00
Jaret Burkett	ead23cee88	Updated supporters	2025-04-05 12:35:01 -06:00
Jaret Burkett	ab59ca5091	Updated comment on control path	2025-04-04 10:23:06 -06:00
Jaret Burkett	eddd3c1611	Added finetuning/training example for redux	2025-04-04 10:05:41 -06:00
Jaret Burkett	b0d0466efd	Add better error messages if name exists when saving a job	2025-04-03 11:25:25 -06:00
Jaret Burkett	ac1ee559c5	Added bluring to mask for flex2	2025-04-02 07:55:51 -06:00
Jaret Burkett	77763a3e5c	Update divisiblity of SD3	2025-04-02 06:49:06 -06:00
Jaret Burkett	a42c5a1de5	Adjust buckets for flex2	2025-04-02 06:47:41 -06:00
Jaret Burkett	3d131fb27a	Added a file signature check on the dataset size caching system to invalidate cached dimensions if the file changes.	2025-04-01 07:39:36 -06:00
Jaret Burkett	5ea19b6292	small bug fixes	2025-03-30 20:09:40 -06:00
Jaret Burkett	58861005a5	Version bump	2025-03-30 09:23:30 -06:00
Jaret Burkett	c083a0e5ea	Allow DFE to not have a VAE	2025-03-30 09:23:01 -06:00
Jaret Burkett	860d892214	Pixel shuffle adapter. Some bug fixes thrown in	2025-03-29 21:15:01 -06:00
Jaret Burkett	b94d7aafea	Have error boundary if simple job cannot be displayed due to the job being advanced	2025-03-27 19:55:40 -06:00
Jaret Burkett	3c95f87a90	Added some missing dependencies	2025-03-27 18:48:48 -06:00
Jaret Burkett	1d5f387f54	Fix docker command to work better with runpod	2025-03-27 17:44:46 -06:00
Jaret Burkett	5365200da1	Added ability to add models to finetune as plugins. Also added flux2 new arch via that method.	2025-03-27 16:07:00 -06:00
Jaret Burkett	e9e30104d3	Merge pull request #271 from ostris/wavelet_loss ... Added experimental wavelet loss	2025-03-26 19:12:09 -06:00
Jaret Burkett	ce4c5291a0	Added experimental wavelet loss	2025-03-26 18:11:23 -06:00
Jaret Burkett	c101f07834	Version bump	2025-03-26 12:16:01 -06:00
Jaret Burkett	e4526ad4a4	Updates to handle video in a dataset on ui	2025-03-26 12:15:28 -06:00
Jaret Burkett	4595965e06	Added an inpainting mask generator for training inpainting if inpaint mask is not provided	2025-03-25 12:16:10 -06:00
Jaret Burkett	41edc18750	Removed unnessary import	2025-03-25 11:54:42 -06:00
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