From ad36c7736d2083ce6644b40f299ada23c366f3cf Mon Sep 17 00:00:00 2001
From: dahoas <alexdahoas@gmail.com>
Date: Thu, 20 Jul 2023 09:24:18 +0000
Subject: [PATCH] Fix: rename model_tok to tokenizer is reward_fn arg

---
 trlx/trainer/accelerate_base_trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/trlx/trainer/accelerate_base_trainer.py b/trlx/trainer/accelerate_base_trainer.py
index cf3b58c5e..d9b7ae8fc 100644
--- a/trlx/trainer/accelerate_base_trainer.py
+++ b/trlx/trainer/accelerate_base_trainer.py
@@ -431,7 +431,7 @@ def evaluate(self):  # noqa: C901
                         samples=str_samples,
                         prompts=str_prompts,
                         outputs=str_outputs,
-                        model_tok=self.tokenizer,
+                        tokenizer=self.tokenizer,
                         **metadata,
                     )
                     if isinstance(rewards[0], torch.Tensor):