bug fixed

40ae4fb9 · u010280923 · f851310a · 40ae4fb9
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

src/dataset.py src/dataset.py +1 -1

未找到文件。
--- a/src/dataset.py
+++ b/src/dataset.py
@@ -305,7 +305,7 @@ class RMDataset(Dataset):

    def __getitem__(self, index):
        ctx_len = self.args.ctx_len
-        req_len = ctx_len + 1
+        req_len = ctx_len
        prompt_prefer_idx, prompt_alter_idx, prompt_prefer_mask, prompt_alter_mask = self.data[index]

        prompt_prefer_idx = prompt_prefer_idx[: req_len]