648 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
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
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