12 Commits

Author	SHA1	Message	Date
Jaret Burkett	79499fa795	Allow fine tuning pruned versions of chroma. Allow flash attention 2 for chroma if it is installed.	2025-05-21 07:02:50 -06:00
Jaret Burkett	6174ba474e	Fixed issue with chroma sampling	2025-05-10 18:30:23 +00:00
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	add83df5cc	Fixed issue with training hidream when batch size is larger than 1	2025-04-21 17:26:29 +00:00
Jaret Burkett	77001ee77f	Upodate model tag on loras	2025-04-19 10:41:27 -06:00
Jaret Burkett	0f99fce004	Adjust hidream lora names to work with comfy	2025-04-16 09:24:23 -06:00
Jaret Burkett	524bd2edfc	Make flash attn optional. Handle larger batch sizes.	2025-04-14 14:34:46 +00: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	7c21eac1b3	Added support for Lodestone Rock's Chroma model	2025-04-05 13:21:36 -06:00