diff --git a/finetune.py b/finetune.py index 742379c..d31a35e 100644 --- a/finetune.py +++ b/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