11 Commits

Author	SHA1	Message	Date
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