fix: filter_fn returns True to quantize, not skip

This commit is contained in:
JWLHS 2026-07-10 12:10:46 +08:00
parent 2d0bb38f4f
commit 015e8a2676

View File

@ -24,7 +24,7 @@ def quantize_model(
model: PyTorch model to quantize.
group_size: Quantization group size (32/64/128/256).
filter_fn: Optional callable(module, name) bool.
Return True to skip that module.
Return True to quantize that module.
Returns:
Same model (quantized in-place).