bug fixed

f851310a · u010280923 · 09fc08b8 · f851310a
显示空白变更内容
内联并排

Showing with 0 addition and 1 deletion

src/rlhf/reward.py src/rlhf/reward.py +0 -1

未找到文件。
--- 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, :]