Skip to content

Commit 98473c6

Browse files
authored
fix: Update DPO and SFT configs to use dtensor (#256)
Signed-off-by: ashors1 <[email protected]>
1 parent 2558444 commit 98473c6

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

examples/configs/dpo.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ policy:
4646
activation_checkpointing_enabled: false
4747

4848
dtensor_cfg:
49-
enabled: false
49+
enabled: true
5050
cpu_offload: False
5151
sequence_parallel: false
5252
activation_checkpointing: false

examples/configs/sft.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ policy:
3333
activation_checkpointing_enabled: false
3434

3535
dtensor_cfg:
36-
enabled: false
36+
enabled: true
3737
cpu_offload: False
3838
sequence_parallel: false
3939
activation_checkpointing: false

0 commit comments

Comments
 (0)