From 200710cf657c6e658eca09711cc7b6ccfaf2100a Mon Sep 17 00:00:00 2001 From: mhh111 Date: Thu, 16 Apr 2026 09:47:23 +0800 Subject: [PATCH] Fix incorrect bias display in extra_repr of Column/RowParallelLinear --- megatron/core/tensor_parallel/layers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/megatron/core/tensor_parallel/layers.py b/megatron/core/tensor_parallel/layers.py index 610700f0a95..2fbe04567b0 100644 --- a/megatron/core/tensor_parallel/layers.py +++ b/megatron/core/tensor_parallel/layers.py @@ -1092,7 +1092,7 @@ def get_extra_state(self) -> None: def extra_repr(self) -> str: """Extra context to add to the module's string representation.""" tp = self.output_size // self.output_size_per_partition - use_bias = self.bias is not None and self.bias is True + use_bias = self.bias is not None return ( f"in_features={self.input_size}, " f"out_features={self.output_size}, " @@ -1354,7 +1354,7 @@ def get_extra_state(self) -> None: def extra_repr(self) -> str: """Extra context to add to the module's string representation.""" tp = self.input_size // self.input_size_per_partition - use_bias = self.bias is not None and self.bias is True + use_bias = self.bias is not None return ( f"in_features={self.input_size}, " f"out_features={self.output_size}, "