Skip to content

Commit 58fdd22

Browse files
Update gated_delta_net.py
1 parent befbcd2 commit 58fdd22

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

megatron/core/ssm/gated_delta_net.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -738,7 +738,7 @@ def get_parameter_local_cp(
738738
slices = [slice(None)] * param.dim()
739739
dim_size = param.size(dim=dim)
740740
slices[dim] = slice(cp_rank * dim_size // cp_size, (cp_rank + 1) * dim_size // cp_size)
741-
param = param[slices]
741+
param = param[tuple(slices)]
742742
return param
743743

744744

0 commit comments

Comments
 (0)