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):
409
409
410
410
if not self .merged :
411
411
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 :
413
414
input_mp = input_mp @ self .lora_A
414
415
input_mp = ReduceScatterOp .apply (input_mp )
415
416
else :
@@ -651,7 +652,8 @@ def forward(self, x: paddle.Tensor):
651
652
652
653
if not self .merged :
653
654
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 :
655
657
input_a = AllGatherOp .apply (input_a )
656
658
delta_mp = (input_a @ self .lora_B ) * self .scaling
657
659
else :
You can’t perform that action at this time.
0 commit comments