diff --git a/vllm_ascend/ops/linear.py b/vllm_ascend/ops/linear.py index eab312d5cf..831d768977 100644 --- a/vllm_ascend/ops/linear.py +++ b/vllm_ascend/ops/linear.py @@ -297,6 +297,7 @@ def forward( self, input_, is_prefill: bool = True, + is_force_scatter: bool = False, ) -> Union[torch.Tensor, tuple[torch.Tensor, Optional[Parameter]]]: if self.custom_op is not None: return self.custom_op.apply(input_)