Update README.md

README.md

@@ -16,7 +16,7 @@ tokenizer.decode(
 model.generate(
     tokenizer.encode(
             "你饿吗？", return_tensors="pt", add_special_tokens=True
-        ), 128)[0],
+        ), max_length = 128)[0],
 skip_special_tokens = True
 ).split("\n-----\n")