Skip to content

Commit a8238d4

Browse files
authored
[training_utils] fix: Using a non-tuple sequence for multidimensional indexing is deprecated (#3314)
1 parent 47a483b commit a8238d4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

verl/utils/ulysses.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ def _pad_tensor(x: Tensor, dim: int, padding_size: int) -> Tensor:
111111
def _unpad_tensor(x: Tensor, dim: int, padding_size: int) -> Tensor:
112112
slc = [slice(None)] * len(x.shape)
113113
slc[dim] = slice(0, -padding_size)
114-
return x[slc]
114+
return x[tuple(slc)]
115115

116116

117117
def slice_input_tensor(x: Tensor, dim: int, padding: bool = True, group: ProcessGroup = None) -> Tensor:
@@ -127,7 +127,7 @@ def slice_input_tensor(x: Tensor, dim: int, padding: bool = True, group: Process
127127
parts = x.size(dim) // sp_world_size
128128
slc = [slice(None)] * len(x.shape)
129129
slc[dim] = slice(sp_rank * parts, (sp_rank + 1) * parts)
130-
return x[slc].contiguous()
130+
return x[tuple(slc)].contiguous()
131131

132132

133133
def all_to_all_tensor(

0 commit comments

Comments
 (0)