ai-toolkit/extensions_built_in/diffusion_models/wan22
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
..
__init__.py Add support for fine tuning Wan 2.2 I2V 14B 2025-08-18 11:43:32 -06:00
wan22_5b_model.py Refactor qwen5b model code to be qwen 5b specific 2025-08-06 10:54:56 -06:00
wan22_14b_i2v_model.py Add support for fine tuning Wan 2.2 I2V 14B 2025-08-18 11:43:32 -06:00
wan22_14b_model.py Fix issue with wan22 14b that woudl load both transformers temporarily resulting in oom on 24GB. 2025-08-28 13:06:31 -06:00
wan22_pipeline.py 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