processing class

huggingface · Oct 7, 2024 · 7c7cca3 · 7c7cca3
1 parent 8186e93
commit 7c7cca3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_rloo_trainer.py b/tests/test_rloo_trainer.py
@@ -112,7 +112,7 @@ def test_rloo_checkpoint(self):
                 policy=self.policy_model,
                 reward_model=self.reward_model,
                 ref_policy=self.policy_ref_model,
-                tokenizer=self.tokenizer,
+                processing_class=self.tokenizer,
                 train_dataset=dummy_dataset,
                 eval_dataset=dummy_dataset,
             )