Update README.md

README.md

@@ -149,13 +149,15 @@ QNetworkGPT2 is an extraordinary AI model that marries Reinforcement Learning (R
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained("ayjays132/QNetworkGPT2")
-
 model = AutoModelForCausalLM.from_pretrained("ayjays132/QNetworkGPT2")
 
-## Initialize a conversation history
+# Set the EOS token as the padding token
+tokenizer.pad_token = tokenizer.eos_token
+
+# Initialize a conversation history
 conversation_history = []
 
-## Start a conversation loop
+# Start a conversation loop
 while True:
     # Get user input
     user_input = input("You: ")