Update README.md

README.md

@@ -32,8 +32,9 @@ model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint,
                                               low_cpu_mem_usage=True,
                                               trust_remote_code=True).to(device)
 
-inputs = tokenizer.encode("def print_hello():", return_tensors="pt").to(device)
-outputs = model.generate(inputs, max_length=12)
+encoding = tokenizer("def print_hello_world():", return_tensors="pt").to(device)
+encoding['decoder_input_ids'] = encoding['input_ids'].clone()
+outputs = model.generate(**encoding, max_length=15)
 print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 ```