mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-07-08 13:57:15 +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
|
assert m.quant_method.quant_config is not None
|
||||||
|
|
||||||
w = m.weight
|
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
|
quant_block_size = m.quant_method.quant_config.weight_block_size
|
||||||
|
|
||||||
assert isinstance(w, torch.Tensor)
|
assert isinstance(w, torch.Tensor)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user