Jaret Burkett a558d5b68f Move transformer back to device on aggresive wan 2.2 pipeline after generation.

2025-07-29 09:13:47 -06:00

..

__init__.py

Add support for Wan2.2 5B

2025-07-29 05:31:54 -06:00

wan22_model.py

Fix bug that prevented training wan 2.2 with batch size greater than 1

2025-07-29 09:06:25 -06:00

wan22_pipeline.py

Move transformer back to device on aggresive wan 2.2 pipeline after generation.

2025-07-29 09:13:47 -06:00