[chore] fix typo (#3)

Author: 1195343015

simumax/core/transformer/language_model.py

@@ -596,4 +596,4 @@ def prefill(self, args, call_stk='', com_buff=None):
         self.call_stk = f"rank{args.rank}-microbatch{args.microbatch}{call_stk}{self.call_stk}"
         for layer in self.children_ordered_module:
             self.layers.append(layer)
-            layer.prefill(args, self.call_stk, com_buff=com_buff)
+            layer.prefill(args, self.call_stk, com_buff=com_buff)