15 Commits

Author	SHA1	Message	Date
Jaret Burkett	76ce757e0c	Added initial support for layer offloading wit Wan 2.2 14B models.	2025-10-20 14:54:30 -06:00
Jaret Burkett	8068755b0a	Fixed issue with wan 2.2 getting stuck on CPU	2025-10-09 17:24:25 +00:00
Jaret Burkett	056711d4ed	Fix issue with wan22 14b that woudl load both transformers temporarily resulting in oom on 24GB.	2025-08-28 13:06:31 -06:00
Jaret Burkett	119653c3f2	Force width, height, and num frames to always be the proper sizes for Wan 2.2 models	2025-08-25 10:33:28 -06:00
Jaret Burkett	d2bbe1872c	Add support for fine tuning Wan 2.2 I2V 14B	2025-08-18 11:43:32 -06:00
Jaret Burkett	b3e666daf4	Fix issue with wan22 14b where timesteps were generated not in the current boundary.	2025-08-16 21:16:48 -06:00
Jaret Burkett	6fffadfc0e	Fixed a bug that prevented training just one stage of Wan 2.2 14b	2025-08-16 18:07:21 -06:00
Jaret Burkett	8ea2cf00f6	Added training to the ui. Still testing, but everything seems to be working.	2025-08-16 05:51:37 -06:00
Jaret Burkett	1c96b95617	Fix issue where sometimes the transformer does not get loaded properly.	2025-08-14 14:24:41 -06:00
Jaret Burkett	3413fa537f	Wan22 14b training is working, still need tons of testing and some bug fixes	2025-08-14 13:03:27 -06:00
Jaret Burkett	14ccf2f3ce	Refactor qwen5b model code to be qwen 5b specific	2025-08-06 10:54:56 -06:00
Jaret Burkett	5d8922fca2	Add ability to designate a dataset as i2v or t2v for models that support it	2025-08-06 09:29:47 -06:00
Jaret Burkett	a558d5b68f	Move transformer back to device on aggresive wan 2.2 pipeline after generation.	2025-07-29 09:13:47 -06:00
Jaret Burkett	1d1199b15b	Fix bug that prevented training wan 2.2 with batch size greater than 1	2025-07-29 09:06:25 -06:00
Jaret Burkett	ca7c5c950b	Add support for Wan2.2 5B	2025-07-29 05:31:54 -06:00