frozenleaves/trl
1
0
Fork 0
mirror of https://github.com/huggingface/trl.git synced 2025-10-20 10:03:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix CI slow test ValueError: Unknown loss type: dapo (#4254)

Browse Source
This commit is contained in:
Albert Villanova del Moral
2025-10-10 16:37:02 +02:00
committed by GitHub
parent f45e86571b
commit 98488e0946
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/slow/test_grpo_slow.py
View File

@ -118,6 +118,7 @@ class TestGRPOTrainerSlow(TrlTestCase):
max_completion_length=self.max_length,
report_to="none",
logging_strategy="no",
loss_type="bnpo", # liger-kernel does not support "dapo" default; see https://github.com/linkedin/Liger-Kernel/issues/620
)
model = AutoModelForCausalLM.from_pretrained(model_name)