681 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Jaret Burkett	43cb5603ad	Added chroma model to the ui. Added logic to easily pull latest, use local, or use a specific version of chroma. Allow ustom name or path in the ui for custom models	2025-05-07 12:06:30 -06:00
Jaret Burkett	d9700bdb99	Added initial support for f-lite model	2025-05-01 11:15:18 -06:00
Jaret Burkett	5890e67a46	Various bug fixes	2025-04-29 09:30:33 -06:00
Jaret Burkett	2b4c525489	Reworked automagic optimizer and did more testing. Starting to really like it. Working well.	2025-04-28 08:01:10 -06:00
Jaret Burkett	88b3fbae37	Various experiments and minor bug fixes for edge cases	2025-04-25 13:44:38 -06:00
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