bug fix

FeixLiu · Oct 22, 2024 · 6df0c68 · 6df0c68
1 parent 76a118b
commit 6df0c68
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddlenlp/transformers/tensor_parallel_utils.py b/paddlenlp/transformers/tensor_parallel_utils.py
@@ -44,7 +44,7 @@ def parallel_matmul(lm_output, logit_weights, tensor_parallel_output=True, train
             logits = xpu_parallel_matmul(
                 lm_output,
                 logit_weights,
-                tensor_parallel_output,
+                tensor_parallel_output=tensor_parallel_output,
                 training=training,
             )
             return logits