Unverified Commit 170e8ea9 authored by Jee Jee Li's avatar Jee Jee Li Committed by GitHub
Browse files

[Misc] Unified linear print info (#23516)


Signed-off-by: default avatarJee Jee Li <pandaleefree@gmail.com>
parent a71e4765
......@@ -1378,7 +1378,7 @@ class RowParallelLinear(LinearBase):
return output, output_bias
def extra_repr(self) -> str:
s = f"input_features={self.input_size_per_partition}"
s = f"in_features={self.input_size_per_partition}"
s += f", output_features={self.output_size}"
s += f", bias={self.bias is not None}"
s += f", tp_size={self.tp_size}"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment