refactor RowParallelLinear

xrsrke · Aug 31, 2023 · d7fb8f2 · d7fb8f2
1 parent a8abb39
commit d7fb8f2
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/pipegoose/nn/tensor_parallel/linear.py b/pipegoose/nn/tensor_parallel/linear.py
@@ -61,10 +61,7 @@ def __init__(
         super().__init__()
         in_per_partition = self._get_input_per_partition(in_features, parallel_context)
 
-        self.in_features = in_features
-        self.out_features = out_features
         self.parallel_context = parallel_context
-
         self.weight = nn.Parameter(torch.randn(out_features, in_per_partition))
 
         if bias is True: