qwix quantize WAN transformer

[susanbao]

Use qwix to quantize the WAN transformer

Add parameters:

• use_qwix_quantization
• quantization_calibration_method

When applying qwix quantization, please set parameters by: use_qwix_quantization=True quantization="fp8_full"

[github-actions]

e2e testgrid: https://8bcf50593faf4ea38060e236169827e5-dot-us-central1.composer.googleusercontent.com/dags/maxdiffusion_tpu_e2e/grid

[coolkp]

Do we need requirements.txt update?

[entrpn]

Can we also add a unit test to verify the quantization logic. Doesn't need to load the full model.

[github-actions]

e2e testgrid: https://8bcf50593faf4ea38060e236169827e5-dot-us-central1.composer.googleusercontent.com/dags/maxdiffusion_tpu_e2e/grid

[susanbao]

Do we need requirements.txt update?

Yes. Just added it.

[susanbao]

Can we also add a unit test to verify the quantization logic. Doesn't need to load the full model.

Have added three unit test on qwix quantization logic.

[coolkp]

I manually tested on runner