mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-24 21:46:21 +08:00
[BugFix] Fix AttributeError: 'MergedColumnParallelLinear' object has no attribute 'weight_scale' (#30399)
Signed-off-by: Lucas Wilkinson <lwilkins@redhat.com>
This commit is contained in:
parent
c756fb6781
commit
aacf0abf8b
@ -89,7 +89,7 @@ def _extract_data_from_linear_base_module(
|
||||
assert m.quant_method.quant_config is not None
|
||||
|
||||
w = m.weight
|
||||
ws = m.weight_scale
|
||||
ws = m.weight_scale_inv if hasattr(m, "weight_scale_inv") else m.weight_scale
|
||||
quant_block_size = m.quant_method.quant_config.weight_block_size
|
||||
|
||||
assert isinstance(w, torch.Tensor)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user