Skip to content

Commit 3f95f13

Browse files
committed
[train]: add dsv4 cicd.
1 parent 2db6879 commit 3f95f13

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/test_utils/config/platforms/cuda.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ a100:
2121
# Add or remove test cases per model
2222
train:
2323
aquila: ["tp2_pp2", "tp4_pp2"]
24-
deepseek: ["tp2_pp2_ep2", "tp2_pp2_ep2_engram"]
24+
deepseek: ["tp2_pp2_ep2", "tp2_pp2_ep2_engram", "tp1_pp2_ep2_v4"]
2525
mixtral: ["tp2_pp1_ep2", "tp4_pp1_ep2"]
2626
hetero_train:
2727
aquila: ["tp2pp1_tp4pp1_tp2pp1", "tp2dp1pp1_tp2dp2pp1_tp1dp2pp1", "dp2dp4_shared_embedding"]

0 commit comments

Comments
 (0)