Here we provide the key implementation of our methods.

We use WAN2.1 model as example.

The Distillation folder is used for distillation in calibration process.

The Reparameterization folder is used for attention reparameterization in inference process.