666 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Jaret Burkett	bd2de5b74e	Remove leco submodule	2025-04-18 10:08:09 -06:00
Jaret Burkett	970fac19a5	Remove batch annotator as submodule	2025-04-18 10:03:37 -06:00
Jaret Burkett	5f312cd46b	Remove ip adapter submodule	2025-04-18 09:59:42 -06:00
Jaret Burkett	c90615f8bb	Add model hooks to polarity loss	2025-04-17 09:00:10 -06:00
Jaret Burkett	5961ef6c9f	Fixed typo in linux install	2025-04-16 22:08:17 -06:00
Jaret Burkett	fd6026ab73	Merge pull request #278 from ostris/hidream ... Add Hidream support	2025-04-16 13:48:27 -06:00
Jaret Burkett	79c87701e7	Add hidream to the ui	2025-04-16 13:45:21 -06:00
Jaret Burkett	fecc64e646	Update hidream defaults, pass additional information to flow guidance	2025-04-16 13:03:04 -06:00
Jaret Burkett	d5a64006b5	Added example config to train hidream	2025-04-16 10:18:22 -06:00
Jaret Burkett	0f99fce004	Adjust hidream lora names to work with comfy	2025-04-16 09:24:23 -06:00
Jaret Burkett	c12036df95	Added ability to use short captions from json caption file	2025-04-16 08:32:28 -06:00
Jaret Burkett	68018c908e	Made a script to convert diffusers to comfy just the transformer	2025-04-15 10:22:05 -06:00
Jaret Burkett	524bd2edfc	Make flash attn optional. Handle larger batch sizes.	2025-04-14 14:34:46 +00:00
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