Update README.md

README.md

@@ -73,6 +73,8 @@ model = AutoModelForCausalLM.from_pretrained(
     device_map="auto",
 )
 tokenizer = AutoTokenizer.from_pretrained(model_name)
+if tokenizer.pad_token is None:
+    tokenizer.pad_token = tokenizer.eos_token
 
 # Initialize conversation context
 initial_messages = [
@@ -118,6 +120,7 @@ while True:
     generated_ids = model.generate(
         **model_inputs,
         max_new_tokens=8192
+        pad_token_id=tokenizer.pad_token_id
     )
 
     # Extract model output, removing special tokens