7 Commits

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