Update tests/test_orpo_trainer.py

Co-authored-by: Quentin Gallouédec <[email protected]>
huggingface · Dec 15, 2024 · 44aa20c · 44aa20c
1 parent b480fff
commit 44aa20c
Showing 1 changed file with 0 additions and 8 deletions.
diff --git a/tests/test_orpo_trainer.py b/tests/test_orpo_trainer.py
@@ -155,14 +155,6 @@ def test_orpo_trainer_with_liger(self):
         with tempfile.TemporaryDirectory() as tmp_dir:
             training_args = ORPOConfig(
                 output_dir=tmp_dir,
-                per_device_train_batch_size=2,
-                max_steps=3,
-                remove_unused_columns=False,
-                gradient_accumulation_steps=1,
-                learning_rate=9e-1,
-                eval_strategy="steps",
-                logging_steps=1,
-                beta=0.1,
                 report_to="none",
                 use_liger_loss=True,  # Enable Liger loss
             )