Update README.md

README.md

@@ -48,7 +48,7 @@ def tokenize_pair(sources:List[str], candidate1s:List[str], candidate2s:List[str
         candidate1_ids = tokenizer.encode(cand1_prefix + candidate1s[i], max_length=candidate_max_length, truncation=True)
         candidate2_ids = tokenizer.encode(cand2_prefix + candidate2s[i], max_length=candidate_max_length, truncation=True)
         ids.append(source_ids + candidate1_ids + candidate2_ids)
-    encodings = tokenizer.pad({"input_ids": ids}, return_tensors="pt", padding=True, max_length=max_length)
+    encodings = tokenizer.pad({"input_ids": ids}, return_tensors="pt", padding="max_length", max_length=max_length)
     return encodings
 
 encodings = tokenize_pair(inputs, candidates_A, candidates_B)