lightx2v

/

Wan2.2-Distill-Models

like 59

Thank you for your excellent work on the wan2.2 model. The standard version of the model works perfectly in my workflow.

I am encountering an issue specifically when trying to load your quantized models (fp8_e4m3 and int8) in ComfyUI. My workflow uses the "Load Diffusion Model" node, but it fails when I select either of the quantized model files.

The console displays the following error message:

unet unexpected: ['blocks.0.self_attn.q.weight_scale', 'blocks.0.self_attn.k.weight_scale', 'blocks.0.self_attn.v.weight_scale', 'blocks.0.self_attn.o.weight_scale', 'blocks.0.cross_attn.q.weight_scale', 'blocks.0.cross_attn.k.weight_scale', ... and many other similar 'weight_scale' keys]
This error suggests that the underlying loading mechanism used by the "Load Diffusion Model" node cannot handle the quantization parameters (weight_scale) present in the model.

Is there a specific custom node or a different loading method required to properly use these quantized models with a node like "Load Diffusion Model"?

Any guidance on how to correctly integrate these models would be very helpful.