소스 검색

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