Skip to content

Commit 87dba55

Browse files
hjh0119Jintao-Huang
authored andcommitted
[grpo] check liger & sp (#4734)
1 parent f4e60f0 commit 87dba55

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

swift/llm/argument/rlhf_args.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -231,6 +231,8 @@ def _check_grpo(self):
231231
assert trl_version >= version.parse('0.18')
232232
if self.delta is not None:
233233
raise ValueError('Liger loss does not support two-sided GRPO loss yet.')
234+
if self.sequence_parallel_size > 1:
235+
raise ValueError('Liger loss does not support sequence parallel yet.')
234236
from trl.import_utils import is_liger_kernel_available
235237
assert is_liger_kernel_available(), (
236238
'Please install/update liger-kernel by running: pip install -U liger-kernel')

0 commit comments

Comments
 (0)