676 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Jaret Burkett	8ff85ba14f	Add Flex2 training example	2025-04-22 11:59:47 -06:00
Jaret Burkett	80f73ce9c0	Update README.md	2025-04-22 10:46:51 -06:00
Jaret Burkett	9f42944056	Update Sponsors	2025-04-22 09:44:56 -06:00
Jaret Burkett	add83df5cc	Fixed issue with training hidream when batch size is larger than 1	2025-04-21 17:26:29 +00:00
Jaret Burkett	12e3095d8a	Fixed issue with saving base model version	2025-04-19 14:34:01 -06:00
Jaret Burkett	77001ee77f	Upodate model tag on loras	2025-04-19 10:41:27 -06:00
Jaret Burkett	d455e76c4f	Cleanup	2025-04-18 11:44:49 -06:00
Jaret Burkett	1628884254	Remove submodule install from docker	2025-04-18 10:41:52 -06:00
Jaret Burkett	9c422ac14f	Bump version	2025-04-18 10:39:51 -06:00
Jaret Burkett	bfe29e2151	Removed all submodules. Submodule free now, yay.	2025-04-18 10:39:15 -06:00
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