File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -409,7 +409,8 @@ def forward(self, x: paddle.Tensor):
409409
410410 if not self .merged :
411411 input_mp = self .lora_dropout (input_mp )
412- if MC2RowSeqParallelCoreLinear is None :
412+ # TODO(@gexiao): temporary workaround for deterministic calculation
413+ if True or MC2RowSeqParallelCoreLinear is None :
413414 input_mp = input_mp @ self .lora_A
414415 input_mp = ReduceScatterOp .apply (input_mp )
415416 else :
@@ -651,7 +652,8 @@ def forward(self, x: paddle.Tensor):
651652
652653 if not self .merged :
653654 input_a = self .lora_dropout (x ) @ self .lora_A
654- if MC2ColumnSeqParallelCoreLinear is None :
655+ # TODO(@gexiao): temporary workaround for deterministic calculation
656+ if True or MC2ColumnSeqParallelCoreLinear is None :
655657 input_a = AllGatherOp .apply (input_a )
656658 delta_mp = (input_a @ self .lora_B ) * self .scaling
657659 else :
You can’t perform that action at this time.
0 commit comments