浏览代码

masking bugfix

Eric Wang 3 年之前
父节点
当前提交
fac53721a2
共有 1 个文件被更改,包括 0 次插入1 次删除
  1. 0 1
      finetune.py

+ 0 - 1
finetune.py

@@ -139,7 +139,6 @@ def generate_and_tokenize_prompt(data_point):
                 user_prompt,
                 truncation=True,
                 max_length=CUTOFF_LEN + 1,
-                padding="max_length",
             )["input_ids"]
         )
         - 1