mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-06 03:20:52 +00:00
[misc] update pre-commit and run all files (#4752)
* [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
This commit is contained in:
@@ -24,7 +24,7 @@ class Dropout(ColossalaiModule):
|
||||
super().__init__(drop, tensor_parallel=tensor_parallel)
|
||||
|
||||
def forward(self, *args):
|
||||
if self.tensor_parallel in [None, '1d']:
|
||||
if self.tensor_parallel in [None, "1d"]:
|
||||
return super().forward(*args)
|
||||
else:
|
||||
with seed(ParallelMode.TENSOR):
|
||||
|
Reference in New Issue
Block a user