From f851310a03c0614276561ea4141642c13f77650d Mon Sep 17 00:00:00 2001
From: u010280923 <xinlu.nlp@foxmail.com>
Date: Fri, 10 Mar 2023 17:26:27 +0800
Subject: [PATCH] bug fixed

---
 src/rlhf/reward.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/src/rlhf/reward.py b/src/rlhf/reward.py
index 3887da2..b9797d1 100644
--- a/src/rlhf/reward.py
+++ b/src/rlhf/reward.py
@@ -171,7 +171,6 @@ class RewardModel(pl.LightningModule):
         # 获得最后一个 token 的 embedding
         last_token_embeds = self.rwkv(
             x,
-            state=None,
             extra_embed=extra_embed,
             rm_train=True
         )[:, -1, :]
-- 
GitLab