From 7cc052f6c0b75c021f4edfcb3f40f54a15e32f26 Mon Sep 17 00:00:00 2001 From: Ofey Chan Date: Thu, 8 Sep 2022 15:22:10 +0800 Subject: [PATCH] [NFC] polish colossalai/nn/layer/colossalai_layer/linear.py (#1556) --- colossalai/nn/layer/colossalai_layer/linear.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/colossalai/nn/layer/colossalai_layer/linear.py b/colossalai/nn/layer/colossalai_layer/linear.py index 35e6a783c..f3f35838b 100644 --- a/colossalai/nn/layer/colossalai_layer/linear.py +++ b/colossalai/nn/layer/colossalai_layer/linear.py @@ -81,7 +81,8 @@ class Linear(ColossalaiModule): else: linear_cls = _parallel_linear[tensor_parallel] gather_output = kwargs.pop('gather_output', None) - if 'gather_output' in inspect.signature(linear_cls.__init__).parameters.keys(): # gather_out arg is available + if 'gather_output' in inspect.signature( + linear_cls.__init__).parameters.keys(): # gather_out arg is available kwargs['gather_output'] = gather_output layer = linear_cls( in_features,