parent
374747b492
commit
f437b36678
1 changed files with 2 additions and 1 deletions
|
|
@ -192,7 +192,8 @@ def is_linear_module(module):
|
||||||
and hasattr(module.quant_method, "quant_config")
|
and hasattr(module.quant_method, "quant_config")
|
||||||
and module.quant_method.quant_config.get_name() == "gptq"):
|
and module.quant_method.quant_config.get_name() == "gptq"):
|
||||||
_USE_VLLM_GPTQ = True
|
_USE_VLLM_GPTQ = True
|
||||||
invalidInputError(module.skip_bias_add is not True, "Currently, ipex-vllm does not"
|
invalidInputError(module.skip_bias_add is not True or module.bias is None,
|
||||||
|
"Currently, ipex-vllm does not"
|
||||||
" support linear layers with skip_bias_add argument")
|
" support linear layers with skip_bias_add argument")
|
||||||
if isinstance(module, RowParallelLinear) and tp_size >= 2:
|
if isinstance(module, RowParallelLinear) and tp_size >= 2:
|
||||||
mp_group = get_tensor_model_parallel_group()
|
mp_group = get_tensor_model_parallel_group()
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue