Wan2.2-I2V-A14B-Moe-Distill-Lightx2v-NVFP4
Overview
This is a
partial NVFP4 quantization of
Wan2.2-I2V-A14B-Moe-Distill-Lightx2v by lightx2v, produced using
convert_to_quant by
silveroxides.
Wan2.2-I2V-A14B-Moe-Distill-Lightx2v is an image-to-video generation model built on
Wan2.2-I2V-A14B. It applies step distillation and classifier-free guidance distillation to reduce inference to
4 steps without CFG, cutting generation time substantially while preserving output quality.
IMPORTANT
Since NVFP4 is only supported on NVIDIA Blackwell architecture GPUs, running this model requires a Blackwell GPU with its corresponding support enabled in torch, along with a recent version of ComfyUI and
comfy-kitchen built against CUDA 13.
Quantization
The model weights have been partially quantized to NVFP4 (NVIDIA Floating Point 4-bit) and MXFP8, quantization formats supported on NVIDIA Blackwell architecture GPUs.
The quantization format assigned to each layer is based on a sensitivity analysis performed with a custom script, which scores each weight tensor using excess kurtosis, dynamic range, and aspect ratio. Thresholds are derived automatically from the model's own score distribution.
The analysis yields the following convert_to_quant parameters. This conversion takes about 4 hours on an RTX 5060 for each model (high and low noise).
1#!/bin/bash
2convert_to_quant -i "${1}" \
3 --nvfp4 --wan --comfy_quant --save-quant-metadata \
4 --custom-type mxfp8 \
5 --custom-layers "blocks\.(1|2|3)\.cross_attn\.k\.weight|blocks\.(6|8|9|10)\.cross_attn\.k\.weight|blocks\.(0|1|2|3)\.cross_attn\.v\.weight|blocks\.(6)\.cross_attn\.q\.weight|blocks\.(6|14)\.cross_attn\.o\.weight|blocks\.(0|1|2|3)\.cross_attn\.v_img\.weight|blocks\.(0)\.self_attn\.k\.weight|blocks\.(7|9|10|12|13|14)\.self_attn\.k\.weight|blocks\.(19)\.self_attn\.q\.weight|blocks\.(0|1|2|3)\.ffn\.0\.weight|blocks\.(36|37|38|39)\.ffn\.0\.weight|blocks\.(39)\.cross_attn\.(k|v|k_img|v_img)\.weight|blocks\.(39)\.self_attn\.(k|v)\.weight" \
6 --exclude-layers "blocks\.(4|5|7)\.cross_attn\.k\.weight|blocks\.(0)\.cross_attn\.q\.weight|blocks\.(5|7|9|10|11|12|19|20)\.cross_attn\.o\.weight|blocks\.(8|11|33)\.self_attn\.k\.weight|blocks\.(38)\.self_attn\.k\.weight|blocks\.(14|16|17)\.self_attn\.q\.weight|blocks\.(39)\.cross_attn\.(q|o)\.weight|blocks\.(39)\.self_attn\.(q|o)\.weight|blocks\.(39)\.ffn\.2\.weight" \
7 --num-iter 6000 \
8 --top-p 0.35 \
9 --calib-samples 8192 \
10 --extract-lora --lora-rank 64 \
11 --lora-target "ffn\.(0|2)\.weight|self_attn\.(v|o)\.weight" \
12 -o "${1%%.safetensors}-nvfp4.safetensors" \
13 --lora-output "${1%%.safetensors}-lora.safetensors"
Two rank-64 LoRAs were also generated that can be used to minimise the effects of the resulting quantization for both high and low models.
Inference
The model can be used in ComfyUI with the following parameters, based on the distilled model's own recommendations:
| Parameter | Value |
|---|
| Shift | 5.0 |
| Sampler | Euler |
| Scheduler | Simple |
| CFG | 1.0 |
| Steps | 4 |
The combinations euler/simple and heun/linear_quadratic (sampler/scheduler) are also known to produce good results.
The model is designed to generate 81 frames and is compatible with LoRAs. Sampling completes in under 60 seconds on an RTX 5060, making it possible to produce a full 81-frame video in under two minutes; with RIFE, those 81 frames convert to a 10-second video.
Abrupt camera movements or fast subject motion may produce artifacts. This is an inherent limitation of applying aggressive quantization to an already distilled model.
License Agreement
This model is licensed under the
Apache 2.0 License. You retain full ownership of your generated content, but are solely responsible for its use in compliance with the license terms and applicable laws.
Acknowledgements
Big kudos to the contributors to the
Wan2.2 and
Self-Forcing repositories for their open research, and to
silveroxides for their quantization tools.
