We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2c5e637 commit d5ca211Copy full SHA for d5ca211
vllm/model_executor/models/baichuan.py
@@ -46,7 +46,7 @@
46
from vllm.model_executor.sampling_metadata import SamplingMetadata
47
from vllm.sequence import IntermediateTensors
48
49
-from .interfaces import SupportsLoRA, SupportsPP
+from .interfaces import SupportsLoRA, SupportsPP, SupportsQuant
50
from .utils import (is_pp_missing_parameter,
51
make_empty_intermediate_tensors_factory, make_layers)
52
@@ -334,7 +334,8 @@ def forward(
334
return hidden_states
335
336
337
-class BaiChuanBaseForCausalLM(nn.Module, SupportsLoRA, SupportsPP):
+class BaiChuanBaseForCausalLM(nn.Module, SupportsLoRA, SupportsPP,
338
+ SupportsQuant):
339
packed_modules_mapping = {
340
"W_pack": ["W_pack"],
341
"gate_up_proj": [
0 commit comments