Update train.py

train.py

@@ -8,6 +8,7 @@ dataset = load_dataset("json", data_files="python.jsonl")
 # Load tokenizer and model
 model_name = "distilgpt2"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
+tokenizer.pad_token = tokenizer.eos_token
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
 # Tokenize function