Update README.md

README.md

@@ -31,8 +31,9 @@ model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint,
                                               torch_dtype=torch.float16,
                                               trust_remote_code=True).to(device)
 
-inputs = tokenizer.encode("def print_hello_world():", return_tensors="pt").to(device)
-outputs = model.generate(inputs, max_length=8)
+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))
 ```