mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-07-18 20:38:15 +08:00
fix: filter_fn returns True to quantize, not skip
This commit is contained in:
parent
2d0bb38f4f
commit
015e8a2676
@ -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).
|
||||
|
||||
Loading…
Reference in New Issue
Block a user